ray/rllib/models/jax
2022-02-22 09:36:44 +01:00
..
__init__.py [RLlib] Add a minimal JAX ModelV2 (FCNet) to RLlib. (#12502) 2020-12-03 15:51:30 +01:00
fcnet.py [CI] Format Python code with Black (#21975) 2022-01-29 18:41:57 -08:00
jax_action_dist.py [CI] Format Python code with Black (#21975) 2022-01-29 18:41:57 -08:00
jax_modelv2.py [CI] Format Python code with Black (#21975) 2022-01-29 18:41:57 -08:00
misc.py [RLlib] Slate-Q tf implementation and tests/benchmarks. (#22389) 2022-02-22 09:36:44 +01:00