cgarciae / nanoGPT-jax

Related projects ⓘ

Alternatives and complementary repositories for nanoGPT-jax

• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆40Updated 2 years ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆49Updated last year
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆53Updated 2 years ago
• epignatelli / discovering-reinforcement-learning-algorithms
  A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…
  ☆20Updated 3 years ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• EmptyJackson / groove
  Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…
  ☆22Updated 4 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• danijar / ninjax
  General Modules for JAX
  ☆59Updated 3 months ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 3 weeks ago
• neuroevobench / neuroevobench
  Neuroevolution Benchmark in JAX 🦕
  ☆37Updated last year
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆102Updated last year
• evgenii-nikishin / omd
  JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
  ☆43Updated 3 years ago
• keraJLi / synthetic-gymnax
  ☆17Updated 5 months ago
• danijar / elements
  Building blocks for productive research
  ☆45Updated last week
• sash-a / CleanRL.jl
  Simple single file implementations of Reinforcement Learning algorithms in Julia
  ☆20Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆34Updated 6 months ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆43Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆62Updated last month
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• mgrankin / minGPT
  minGPT in JAX
  ☆46Updated 2 years ago
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆60Updated last year
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆9Updated last year
• roger-creus / Wave-Defense-Learning-Environment
  A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.
  ☆14Updated last year
• google-deepmind / neural_testbed
  ☆191Updated 6 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆127Updated 3 weeks ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆89Updated 11 months ago
• DramaCow / jaxued
  ☆63Updated 3 months ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆54Updated 2 years ago