7 projects
Navix
Accelerated gridworld navigation with JAX for deep reinforcement learning
HELX
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
HELX-agents
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
HELX-envs
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
HELX-base
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
jax-enums
JAX-compatible Enumerations.
jaxboard
A tensorboard logger for jax