pure-jax

There are 0 repository under pure-jax topic.

EdanToledo / Stoix
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
anakin deep-reinforcement-learning distributed-reinforcement-learning end-to-end jax pure-jax reinforcement-learning reinforcement-learning-algorithms research sarl sebulba single-agent single-agent-reinforcement-learning single-agent-systems
Language:Python 209