Alternatives and similar repositories for kauldron

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

• google / qwix
  a Jax quantization library
  ☆83Updated this week
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆97Updated 5 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆297Updated last year
• AI-Hypercomputer / maxdiffusion
  ☆294Updated this 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.
  ☆155Updated 2 months ago
• google / praxis
  ☆191Updated 3 weeks ago
• google-deepmind / nanodo
  ☆287Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆556Updated 3 weeks ago
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆225Updated last week
• AI-Hypercomputer / RecML
  ☆214Updated this week
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆360Updated last week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆300Updated last year
• apple / ml-sigma-reparam
  ☆314Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 4 months ago
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆177Updated last year
• microsoft / dion
  Dion optimizer algorithm
  ☆413Updated this week
• google-deepmind / tf2jax
  ☆118Updated last month
• m-a-n-i-f-e-s-t / power-attention
  Attention Kernels for Symmetric Power Transformers
  ☆128Updated 3 months ago
• NVlabs / RLP
  RLP: Reinforcement as a Pretraining Objective
  ☆222Updated 3 months ago
• google / grain
  Library for reading and processing ML training data.
  ☆643Updated last week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆373Updated this week
• google-deepmind / mishax
  ☆150Updated 4 months ago
• google-deepmind / regress-lm
  Library for text-to-text regression, applicable to any input string representation and allows pretraining and fine-tuning over multiple r…
  ☆305Updated 3 weeks ago
• apple / ml-pararnn
  ☆158Updated 2 months ago
• erfanzar / EasyDeL
  Accelerate, Optimize performance with streamlined training and serving options with JAX.
  ☆328Updated last week
• CLAIRE-Labo / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆121Updated last month
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆169Updated 5 months ago
• epfml / DenseFormer
  ☆82Updated last year
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆133Updated 3 weeks ago