jaxrl
indaba-pracs-2022
| jaxrl | indaba-pracs-2022 | |
|---|---|---|
| 2 | 1 | |
| 576 | 172 | |
| - | 0.6% | |
| 0.0 | 0.0 | |
| over 1 year ago | about 1 month ago | |
| Jupyter Notebook | Jupyter Notebook | |
| MIT License | Apache License 2.0 |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
jaxrl
-
JAX in Reinforcement Learning
Have you looked at this repo or this repo ?
-
CleanRL now has a DDPG + JAX implementation roughly 2.5-4x faster than DDPG + PyTorch
https://github.com/ikostrikov/jaxrl would be another great reference implementation. Probably you want to also checkout the docs for jax, flax, and optax.
indaba-pracs-2022
-
From Deep Learning Foundations to Stable Diffusion
This year's Deep Learning Indaba had a tutorial on diffusion models in Jax: https://github.com/deep-learning-indaba/indaba-pracs-2022/tr...
What are some alternatives?
jaxrl_m - Skeleton for scalable and flexible Jax RL implementations
PyCBC-Tutorials - Learn how to use PyCBC to analyze gravitational-wave data and do parameter inference.
sbx - SBX: Stable Baselines Jax (SB3 + Jax)
neural-tangents - Fast and Easy Infinite Neural Networks in Python
cleanrl - High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
pymc-resources - PyMC educational resources
Popular-RL-Algorithms - PyTorch implementation of Soft Actor-Critic (SAC), Twin Delayed DDPG (TD3), Actor-Critic (AC/A2C), Proximal Policy Optimization (PPO), QT-Opt, PointNet..
bodywork-pymc3-project - Serving Uncertainty with Bayesian inference, using PyMC3 with Bodywork
uvadlc_notebooks - Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
brax - Massively parallel rigidbody physics simulation on accelerator hardware.
nn - 🧑🏫 60 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠