Alternatives and similar repositories for milabench:

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

• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆48Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• google-deepmind / tf2jax
  ☆114Updated last week
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• srush / Tensor-Puzzles-Penzai
  ☆20Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 7 months ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆172Updated last week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated 10 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• jax-ml / jax-llm-examples
  ☆102Updated this week
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 5 months ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆57Updated 2 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆92Updated last year
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆194Updated 2 months ago
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆48Updated last year
• lebrice / torch_jax_interop
  Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
  ☆27Updated 2 weeks ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆31Updated 10 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 5 months ago
• jax-ml / jax-ai-stack
  ☆157Updated this week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆94Updated 4 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆31Updated 3 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 5 months ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆59Updated 3 years ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 4 months ago
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 9 months ago