Alternatives and similar repositories for nanoGPT-jax

Users that are interested in nanoGPT-jax are comparing it to the libraries listed below

• jenkspt / gpt-jax

  View on GitHub
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆63Feb 15, 2023Updated 3 years ago
• mlpc-ucsd / XTRA

  View on GitHub
  On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning
  ☆16Apr 30, 2023Updated 2 years ago
• araffin / datasaurust

  View on GitHub
  Blazingly fast implementation of the Datasaurus paper. Same Stats, Different Graphs.
  ☆19Apr 10, 2023Updated 2 years ago
• roger-creus / Wave-Defense-Learning-Environment

  View on GitHub
  A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.
  ☆15Jan 3, 2023Updated 3 years ago
• XinJingHao / Actor-Sharer-Learner

  View on GitHub
  Actor-Sharer-Learner training framework for off-policy DRL algorithms
  ☆22Dec 29, 2024Updated last year
• hlml / fortuitous_forgetting

  View on GitHub
  ☆19Apr 16, 2022Updated 3 years ago
• Howuhh / prioritized_experience_replay

  View on GitHub
  Prioritized Experience Replay implementation with proportional prioritization
  ☆86Jul 18, 2023Updated 2 years ago
• AllanYangZhou / midGPT

  View on GitHub
  Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.
  ☆24Sep 29, 2024Updated last year
• llm-fireq / fireq

  View on GitHub
  FireQ: Fast INT4-FP8 Kernel and RoPE-aware Quantization for LLM Inference Acceleration
  ☆20Jun 27, 2025Updated 7 months ago
• ikostrikov / dmcgym

  View on GitHub
  ☆23Aug 19, 2022Updated 3 years ago
• Hwhitetooth / jax_muzero

  View on GitHub
  An implementation of MuZero in JAX.
  ☆57Nov 8, 2022Updated 3 years ago
• theogruner / rl_pro_telu

  View on GitHub
  ☆23Jun 8, 2021Updated 4 years ago
• AakashKumarNain / mistral_jax

  View on GitHub
  This is a port of Mistral-7B model in JAX
  ☆33Jul 1, 2024Updated last year
• FLAIROx / JaxGL

  View on GitHub
  Simple JAX Graphics Library.
  ☆36Nov 3, 2024Updated last year
• young-geng / mintext

  View on GitHub
  Minimal but scalable implementation of large language models in JAX
  ☆35Nov 28, 2025Updated 2 months ago
• ymd-h / cpprb

  View on GitHub
  Fast Flexible Replay Buffer Library (Mirror repository of https://gitlab.com/ymd_h/cpprb)
  ☆74Dec 14, 2024Updated last year
• tinkoff-ai / katakomba

  View on GitHub
  Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)
  ☆76Jun 23, 2023Updated 2 years ago
• Raffi-Hagopian / AI-CAC

  View on GitHub
  AI-CAC is a deep learning model that segments and scores coronary artery calcium (CAC) on routine non-gated, non-contrast chest CT scans.
  ☆19Jan 29, 2026Updated 2 weeks ago
• kaloureyes3 / v4-clients

  View on GitHub
  ☆10Apr 5, 2024Updated last year
• jolle-ag / qdx

  View on GitHub
  Quantum error correction code AI-discovery with Jax
  ☆47May 26, 2025Updated 8 months ago
• seohongpark / HIQL

  View on GitHub
  HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)
  ☆93Dec 1, 2024Updated last year
• thanhtbt / ROLCP

  View on GitHub
  [IEEE ICASSP 2021] "A fast randomized adaptive CP decomposition for streaming tensors". In 46th IEEE International Conference on Acoustic…
  ☆11Feb 16, 2023Updated 3 years ago
• Columbia-NLP-Lab / LionAlignment

  View on GitHub
  ☆12Aug 6, 2024Updated last year
• MIAGroupUT / carotid-segmentation

  View on GitHub
  The source code of the winning algorithm of the MICCAI Grand Challenge Carotid Artery Vessel Wall Segmentation Challenge
  ☆13Apr 15, 2024Updated last year
• facebookresearch / dmae_st

  View on GitHub
  Directed masked autoencoders
  ☆14Feb 5, 2026Updated last week
• bishopAL / GeRot

  View on GitHub
  ☆11Sep 26, 2022Updated 3 years ago
• endiba360 / compiler-in-python

  View on GitHub
  A Compiler made with python using some useful libreries.
  ☆11Nov 11, 2019Updated 6 years ago
• jax-ml / bonsai

  View on GitHub
  Minimal, lightweight JAX implementations of popular models.
  ☆191Feb 6, 2026Updated last week
• google-deepmind / nanodo

  View on GitHub
  ☆292Jul 15, 2024Updated last year
• yardenas / la-mbda

  View on GitHub
  LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization
  ☆38Jan 16, 2023Updated 3 years ago
• hari-sikchi / LOOP

  View on GitHub
  Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]
  ☆41Aug 27, 2022Updated 3 years ago
• hr0nix / dejax

  View on GitHub
  Accelerated replay buffers in JAX
  ☆46Sep 17, 2022Updated 3 years ago
• davidar / eigenGPT

  View on GitHub
  Minimal C++ implementation of GPT2
  ☆41Jul 1, 2023Updated 2 years ago
• iCGY96 / awesome_concept_learning_list

  View on GitHub
  A curated list of papers & resources linked to concept learning
  ☆12Aug 9, 2023Updated 2 years ago
• Na-VAE / dino_perceptual

  View on GitHub
  DINO-based perceptual losses and FDD feature extraction
  ☆24Jan 7, 2026Updated last month
• U-C4N / Umap

  View on GitHub
  Umap is a Python library that transforms OpenStreetMap data into customized maps with minimal code. Create minimalist or multi-layered ma…
  ☆13Feb 9, 2026Updated last week
• vwxyzjn / SC2AI

  View on GitHub
  Integrated Tensorforce and OpenAI Gym to train SC II game agents.
  ☆13Oct 26, 2019Updated 6 years ago
• rikhuijzer / Skans.jl

  View on GitHub
  Monitor web pages and get notified when a page has changed
  ☆12Dec 2, 2022Updated 3 years ago
• rockinbinbin / nlp-emoji-predictor

  View on GitHub
  😂 Predicts an emoji based on sentiment & semantic analyses for a test dataset based on a training dataset of tweets.
  ☆10Mar 17, 2017Updated 8 years ago