Alternatives and similar repositories for milabench

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

• google-deepmind / tf2jax
  ☆114Updated last week
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆139Updated 8 months ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆49Updated 2 years ago
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆62Updated 4 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆106Updated 7 months ago
• jax-ml / jax-llm-examples
  ☆135Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆71Updated 2 weeks ago
• enajx / HyperNCA
  ☆39Updated 3 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 8 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆206Updated 5 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 3 weeks ago
• NERSC / slurm-ray-cluster
  ☆58Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆281Updated last week
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆188Updated this week
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆32Updated last year
• google / init2winit
  ☆76Updated last week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆187Updated last week
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• google-research / jaxpruner
  ☆230Updated 5 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆41Updated 7 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆55Updated 2 months ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆59Updated 2 years ago