9 Commits

Author	SHA1	Message	Date
Sven Mika	8726521604	[RLlib] JAXPolicy prep PR #2 (move get_activation_fn (backward-compatibly), minor fixes and preparations). (#13091)	2020-12-30 22:30:52 -05:00
Sven Mika	d5604eaba3	[RLlib] Attention nets PyTorch support and cleanup (using traj. view API). (#12029)	2020-12-21 18:38:34 -08:00
Sven Mika	b2bcab711d	[RLlib] Attention Nets: tf (#12753)	2020-12-20 20:22:32 -05:00
Michael Luo	b2984d1c34	[RLlib] Model Annotations to Torch Models (#9749)	2020-11-12 12:16:12 +01:00
Sven Mika	d9f1874e34	[RLlib] Minor fixes (torch GPU bugs + some cleanup). (#11609)	2020-10-27 10:00:24 +01:00
Sven Mika	d3bc20b727	[RLlib] ConvTranspose2D module (#11231)	2020-10-12 15:00:42 +02:00
Barak Michener	8e76796fd0	ci: Redo format.sh --all script & backfill lint fixes (#9956)	2020-08-07 16:49:49 -07:00
Sven Mika	5dc4b6686e	[RLlib] Implement DQN PyTorch distributional head. (#9589)	2020-07-25 09:29:24 +02:00
Tanay Wakhare	f77c638d6d	Pytorch AttentionNet (#9088)	2020-06-23 20:42:30 +02:00