Pinned

1. flash-nanoGPT flash-nanoGPT Public

   Jax/Flax re-write of @karpathy 🐐 NanoGPT using some of the common Jax libraries/features (shmap, pallas, jmp, optax, orbax)

   Python 1 1

2. Highway-PPO-Agent-in-Jax Highway-PPO-Agent-in-Jax Public

   a jax implementation of highway environment PPO agent with discrete and continuous spaces

   Python 1

3. RL-environment-in-Jax RL-environment-in-Jax Public

   an implementation of an RL environment in Jax

   Jupyter Notebook 1

4. Imapal-Cartpole-Agent-in-Jax Imapal-Cartpole-Agent-in-Jax Public

   training script for CartPole Impala agent in Jax. the code is inspired from the official Haiku repository

   Jupyter Notebook

5. state-of-the-art-DeepRL state-of-the-art-DeepRL Public

   an implementation of state of the art deep reinforcement learning algorithms benchmarked using open gym ai environments

   Python 2 1

6. Quaada-TensorFlow Quaada-TensorFlow Public

   Introduction to TenorFlow 2.0 with a series of Jupyter Notebook