7 commits

Author	SHA1	Message	Date
Sven Mika	e973b726c2	[RLlib] Support native tf.keras.Models (part 2) - Default keras models for Vision/RNN/Attention. (#15273)	2021-04-30 19:26:30 +02:00
Sven Mika	9eba1871bb	[RLlib] Support easy use_attention=True flag for using the GTrXL model. (#11698)	2021-01-01 14:06:23 -05:00
Sven Mika	b2bcab711d	[RLlib] Attention Nets: tf (#12753)	2020-12-20 20:22:32 -05:00
Sven Mika	3f4bc16276	[RLlib] Add a minimal JAX ModelV2 (FCNet) to RLlib. (#12502)	2020-12-03 15:51:30 +01:00
Sven Mika	2589309cf0	[RLlib] Make sure torch and tf behave the same wrt conv2d nets. (#8785)	2020-06-20 00:05:19 +02:00
Sven Mika	2746fc0476	[RLlib] Auto-framework, retire use_pytorch in favor of framework=... (#8520)	2020-05-27 16:19:13 +02:00
Sven Mika	0422e9c5a8	[RLlib] Add 2 Transformer learning test cases on StatelessCartPole (PPO and IMPALA). (#8624)	2020-05-27 10:19:47 +02:00