Alternatives and similar repositories for trainit:

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

• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆34Updated 3 weeks ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 4 months ago
• f-dangel / singd
  [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)
  ☆21Updated 4 months ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆122Updated 11 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆171Updated this week
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆176Updated this week
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆30Updated 2 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆77Updated 5 months ago
• GallagherCommaJack / modulax
  ☆17Updated 7 months ago
• nikhilvyas / SOAP
  ☆171Updated 3 months ago
• fKunstner / noise-sgd-adam-sign
  ☆16Updated last year
• acerbilab / amortized-conditioning-engine
  Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
  ☆13Updated last week
• shikaiqiu / compute-better-spent
  ☆52Updated 5 months ago
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆60Updated 4 months ago
• AndPotap / einsum-search
  ☆30Updated 5 months ago
• google-deepmind / spectral_ssm
  ☆31Updated 11 months ago
• opooladz / Preconditioned-Stochastic-Gradient-Descent
  A repo based on XiLin Li's PSGD repo that extends some of the experiments.
  ☆14Updated 5 months ago
• jongharyu / neural-svd
  ☆16Updated 6 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆57Updated 4 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 3 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 4 months ago
• locuslab / edge-of-stability
  ☆65Updated 3 months ago
• epfml / REQ
  ☆17Updated 9 months ago
• MarlonBecker / MSAM
  ☆19Updated last year
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆62Updated last month
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆60Updated last year
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆41Updated 2 months ago
• jax-ml / jax-llm-examples
  ☆87Updated 2 weeks ago