Alternatives and similar repositories for kauldron:

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

• google-deepmind / tf2jax
  ☆112Updated last month
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆34Updated 2 months ago
• apple / ml-ademamix
  ☆57Updated 3 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 9 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆133Updated 4 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 2 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 9 months ago
• leloykun / modded-nanogpt
  NanoGPT (124M) quality in 2.67B tokens
  ☆28Updated 2 weeks ago
• google-deepmind / asyncdiloco
  ☆43Updated last year
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆29Updated 4 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 4 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• cloneofsimo / zeroshampoo
  ☆33Updated 6 months ago
• epfml / DenseFormer
  ☆78Updated 10 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆230Updated 3 weeks ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆107Updated 2 months ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆165Updated this week
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆50Updated 8 months ago
• cloneofsimo / min-fsdp
  ☆75Updated 8 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 8 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆76Updated 4 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 5 months ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆17Updated last month
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆28Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 3 months ago
• google-deepmind / einshape
  ☆101Updated 8 months ago