purejaxrl
by luchris429
Pythonpushed about 2 years ago
Really Fast End-to-End Jax RL Implementations
AI summary
RL framework
A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality
- stars
- 755
- forks
- 63
- watching
- 12
- awesome list
- 1
Featured in 1 awesome list
Each link jumps to the spot where the list mentions purejaxrl.