Alternatives and similar repositories for milabench

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

• google-deepmind / tf2jax
  ☆118Updated this week
• NERSC / slurm-ray-cluster
  ☆61Updated 2 years ago
• marin-community / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆212Updated last month
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 5 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated 10 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last month
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆115Updated 11 months ago
• google / init2winit
  ☆79Updated last week
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆152Updated 3 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• google-research / jaxpruner
  ☆234Updated 10 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆202Updated this week
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆54Updated 2 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• davisyoshida / qax
  If it quacks like a tensor...
  ☆59Updated last year
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆103Updated this week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• google-deepmind / jaxline
  ☆161Updated 2 years ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆52Updated last year
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆207Updated last week
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆367Updated this week
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆38Updated 3 years ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆48Updated 11 months ago