Alternatives and similar repositories for kauldron

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

• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆96Updated 3 months ago
• google / qwix
  a Jax quantization library
  ☆53Updated this week
• google-deepmind / nanodo
  ☆283Updated last year
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆187Updated last month
• microsoft / dion
  Dion optimizer algorithm
  ☆374Updated last month
• kvfrans / splus
  ☆120Updated 4 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last year
• openai / chz
  ☆179Updated 2 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆295Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆299Updated 2 months ago
• AI-Hypercomputer / maxdiffusion
  ☆269Updated last week
• AI-Hypercomputer / RecML
  ☆211Updated last week
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆147Updated 6 months ago
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆172Updated last year
• iliao2345 / CompressARC
  ☆197Updated 2 months ago
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆298Updated this week
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Updated 10 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆185Updated 9 months ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase
  ☆77Updated this week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆294Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆542Updated this week
• apoorvkh / academic-pretraining
  $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources
  ☆147Updated last month
• erfanzar / EasyDeL
  Accelerate, Optimize performance with streamlined training and serving options with JAX.
  ☆317Updated this week
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆54Updated last month
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆103Updated 3 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆356Updated this week
• epfml / DenseFormer
  ☆81Updated last year
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆330Updated this week
• CLAIRE-Labo / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆116Updated last week