Alternatives and similar repositories for carbs:

Users that are interested in carbs are comparing it to the libraries listed below

• google-deepmind / nanodo
  ☆199Updated 5 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆174Updated 3 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆136Updated 5 months ago
• keraJLi / rejax
  ☆158Updated this week
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆27Updated last month
• KellerJordan / Muon
  Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead
  ☆174Updated this week
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆468Updated last month
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆85Updated 3 weeks ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆68Updated last week
• balrog-ai / BALROG
  Benchmarking Agentic LLM and VLM Reasoning On Games
  ☆74Updated last week
• modula-systems / modula
  🧱 Modula software package
  ☆130Updated 3 weeks ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆122Updated 4 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆128Updated last month
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆216Updated 2 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 7 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆106Updated 3 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆113Updated 8 months ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆127Updated 2 weeks ago
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆114Updated 6 months ago
• lowrollr / turbozero
  fast + parallel AlphaZero in JAX
  ☆87Updated 3 weeks ago
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆526Updated this week
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆42Updated 5 months ago
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆251Updated last week
• Reytuag / transformerXL_PPO_JAX
  ☆68Updated last month
• Aleph-Alpha / scaling
  Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…
  ☆53Updated last month
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆136Updated 3 weeks ago
• DramaCow / jaxued
  ☆65Updated 3 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆134Updated 9 months ago
• natolambert / rlhf-book
  Textbook on reinforcement learning from human feedback
  ☆78Updated last month