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.
  ☆49Updated last year
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆49Updated last year
• google-deepmind / tf2jax
  ☆114Updated this week
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆137Updated 7 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆177Updated last month
• NERSC / slurm-ray-cluster
  ☆55Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 5 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 6 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated 11 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 11 months ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated last year
• jax-ml / jax-llm-examples
  ☆118Updated 2 weeks ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 7 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆106Updated 10 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆91Updated last year
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆31Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated 10 months ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated 2 years ago
• srush / Tensor-Puzzles-Penzai
  ☆20Updated last year
• 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
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆199Updated 4 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated last week