Alternatives and similar repositories for dit_flow

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

• opooladz / Preconditioned-Stochastic-Gradient-Descent
  A repo based on XiLin Li's PSGD repo that extends some of the experiments.
  ☆14Updated 9 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆35Updated 9 months ago
• jax-ml / jax-llm-examples
  ☆141Updated this week
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• kvfrans / splus
  ☆115Updated last month
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆35Updated last year
• NX-AI / xlstm-jax
  Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…
  ☆98Updated 6 months ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆19Updated 2 weeks ago
• GallagherCommaJack / modulax
  ☆17Updated 11 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆140Updated 9 months ago
• nikhilvyas / SOAP
  ☆206Updated 8 months ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆59Updated 2 years ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆36Updated last week
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆79Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆85Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆34Updated 7 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆140Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated 7 months ago
• apple / ml-ademamix
  ☆65Updated 8 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆252Updated last week
• microsoft / dion
  Dion optimizer algorithm
  ☆193Updated this week
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆94Updated 2 months ago
• radarFudan / mamba-minimal-jax
  ☆31Updated 8 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 7 months ago
• google-deepmind / spectral_ssm
  ☆33Updated last year
• kvfrans / lmpo
  ☆95Updated 3 weeks ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆89Updated last year
• Reytuag / transformerXL_PPO_JAX
  ☆81Updated 9 months ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago