Alternatives and similar repositories for BabyTorch:

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

• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆272Updated 10 months ago
• NX-AI / xlstm-jax
  Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…
  ☆91Updated 3 months ago
• google-deepmind / nanodo
  ☆216Updated 9 months ago
• nikhilvyas / SOAP
  ☆175Updated 4 months ago
• Jaykef / Triton-nanoGPT
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆18Updated 6 months ago
• BlackHC / neural_net_checklist
  ☆150Updated 8 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆189Updated this week
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆154Updated 2 weeks ago
• jax-ml / jax-llm-examples
  ☆102Updated this week
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆134Updated 8 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆89Updated 3 weeks ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆181Updated 8 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆120Updated 8 months ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆51Updated 9 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆188Updated 3 weeks ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 5 months ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆178Updated this week
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆105Updated 5 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆80Updated 4 months ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆57Updated this week
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆51Updated last year
• okarthikb / state-space-models
  ☆27Updated 9 months ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆82Updated last year
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆150Updated 9 months ago
• cloneofsimo / min-fsdp
  ☆78Updated 9 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆180Updated 7 months ago
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆94Updated 3 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆87Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆81Updated last month