#

pure-jax

Here is 1 public repository matching this topic...

EdanToledo / Stoix

🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL

research reinforcement-learning end-to-end deep-reinforcement-learning reinforcement-learning-algorithms sarl jax distributed-reinforcement-learning anakin single-agent sebulba single-agent-systems single-agent-reinforcement-learning pure-jax

Updated Dec 4, 2024
Python

Improve this page

Add a description, image, and links to the pure-jax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the pure-jax topic, visit your repo's landing page and select "manage topics."