jaxrl
uvadlc_notebooks
jaxrl | uvadlc_notebooks | |
---|---|---|
2 | 2 | |
576 | 2,151 | |
- | - | |
0.0 | 6.6 | |
over 1 year ago | 12 days ago | |
Jupyter Notebook | Jupyter Notebook | |
MIT License | MIT License |
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.
uvadlc_notebooks
-
I miss the old days where people asked me to recreate “Facebook” or “Twitter”
So, I don’t have anything simple that’s readily available, and I don’t know how much you’d get from the code itself without some background. But I would recommend the UVA Deep Learning tutorials. Particularly, I’d recommend trying the autoencoder as a good start (tutorial 9). Autoencoders are very easy and fast models to train.
-
[D] Favorite Colab Notebooks / runnable tutorials on adversarial CV
DL course from the University of Amsterdam:Github and Colab including another FGSM example
What are some alternatives?
jaxrl_m - Skeleton for scalable and flexible Jax RL implementations
tensor-sensor - The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
sbx - SBX: Stable Baselines Jax (SB3 + Jax)
hyper-nn - Easy Hypernetworks in Pytorch and Jax
indaba-pracs-2022 - Notebooks for the Practicals at the Deep Learning Indaba 2022.
pyprobml - Python code for "Probabilistic Machine learning" book by Kevin Murphy
cleanrl - High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
get-started-with-JAX - The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
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..
TF_JAX_tutorials - All about the fundamental blocks of TF and JAX!
labml - 🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱