Alternatives and similar repositories for acco

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

• nikhilvyas / SOAP
  ☆234Updated last year
• google / grain
  Library for reading and processing ML training data.
  ☆647Updated this week
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆365Updated this week
• artefactory / woodtapper
  WoodTapper — a Python toolbox for interpretable and explainable tree ensembles.
  ☆22Updated last month
• srush / annotated-s4
  Implementation of https://srush.github.io/annotated-s4
  ☆510Updated 6 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆280Updated 3 weeks ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆556Updated 3 weeks ago
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆474Updated this week
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆459Updated 9 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 4 months ago
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆183Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 3 months ago
• jax-ml / jax-ai-stack
  ☆261Updated 2 weeks ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• google-deepmind / nanodo
  ☆287Updated last year
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…
  ☆292Updated this week
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆80Updated last year
• google-research / jaxpruner
  ☆233Updated 11 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆98Updated 3 weeks ago
• haydn-jones / SOAP_JAX
  Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)
  ☆22Updated this week
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated last year
• sail-sg / stde
  Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
  ☆125Updated last year
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆61Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated last month
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆188Updated 2 weeks ago
• marin-community / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆688Updated 2 weeks ago
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆304Updated this week
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆371Updated 3 years ago
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆226Updated last week
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆210Updated 11 months ago