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
• google-deepmind / tf2jax
  ☆112Updated last month
• NERSC / slurm-ray-cluster
  ☆55Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated 11 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 3 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 4 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 4 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated 9 months ago
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆47Updated last year
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆167Updated this week
• srush / Tensor-Puzzles-Penzai
  ☆20Updated 11 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆90Updated 3 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆90Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• google-research / jaxpruner
  ☆220Updated last month
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆193Updated last month
• instadeepai / outer-value-function-meta-rl
  Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function
  ☆13Updated 2 years ago
• enajx / HyperNCA
  ☆37Updated 2 years ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆172Updated this week
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• jax-ml / jax-ai-stack
  ☆141Updated last week
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆31Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 9 months ago
• srush / drop7
  ☆18Updated 11 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆29Updated 2 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated this week