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
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 7 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• google / drjax
  ☆13Updated last week
• 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
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆139Updated 7 months ago
• google-deepmind / tf2jax
  ☆114Updated this week
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆47Updated 6 months ago
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆49Updated last year
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆31Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆93Updated last week
• jax-ml / jax-llm-examples
  ☆126Updated last month
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• distdl / distdl
  ☆21Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆70Updated 3 weeks ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 3 weeks ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• srush / Tensor-Puzzles-Penzai
  ☆20Updated last year
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆181Updated this week
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆114Updated this week
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 6 months ago
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆24Updated last year
• google / init2winit
  ☆76Updated this week
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆162Updated 2 weeks ago
• cgarciae / einop
  ☆60Updated 3 years ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated last year
• Artur-Galstyan / statedict2pytree
  ☆43Updated 3 weeks ago