A JAX-native High Performance Eval Metrics Library
☆64Aug 31, 2026Updated this week
Alternatives and similar repositories for metrax
Users that are interested in metrax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- a Jax quantization library☆131Updated this week
- Minimal, lightweight JAX implementations of popular models.☆242May 29, 2026Updated 3 months ago
- ☆311Updated this week
- Library for reading and processing ML training data.☆774Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆534Updated this week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated last year
- Coordinates in JAX☆37Updated this week
- Minimal yet performant LLM examples in pure JAX☆277Aug 23, 2026Updated last week
- Tokamax: A GPU and TPU kernel library.☆280Updated this week
- A Lightweight LLM Post-Training Library☆2,427Updated this week
- ☆19Updated this week
- Performing parameter estimation on gravitational wave data with machine learning☆21May 4, 2026Updated 4 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆66Jan 23, 2026Updated 7 months ago
- Package of Pathways-on-Cloud utilities☆33Updated this week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code for the ICML 2023 paper "Geometric Autoencoders - What You See is What You Decode"☆17Jul 28, 2023Updated 3 years ago
- Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡☆36Updated this week
- Reproduce analyses and baselines from the State paper☆17Jun 29, 2026Updated 2 months ago
- Special functions in Jax☆18Mar 1, 2025Updated last year
- Deep memory and sequence models in JAX☆32Jun 8, 2026Updated 2 months ago
- JAX-Toolbox☆428Updated this week
- Stochastic trace estimation using JAX☆21Aug 27, 2026Updated last week
- ☆385Updated this week
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆21Apr 20, 2023Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Reinforcement learning with Equinox☆21Mar 4, 2025Updated last year
- Evaluation suite for transcriptomic perturbation effect prediction models. Includes support for single-cell foundation models.☆41Jul 29, 2025Updated last year
- Investigating the role of pre-training dataset size and diversity on the performance of single-cell foundation models☆21May 12, 2026Updated 3 months ago
- torch sparse linear algebra library☆57Jul 10, 2026Updated last month
- A simple, performant and scalable JAX-based world modeling codebase.☆163Jan 15, 2026Updated 7 months ago
- pytorch dataloaders for single-cell perturbation data☆51Jul 18, 2026Updated last month
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 8 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆21Updated this week
- ☆38Oct 31, 2025Updated 10 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆1,405Updated this week
- Torch utilities for doing machine learning in gravitational wave physics☆34Updated this week
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- ☆18May 20, 2026Updated 3 months ago
- Fast and reliable distributed systems in Python☆35Mar 25, 2026Updated 5 months ago
- A Python library for defining structured command-line flags.☆34Jun 10, 2026Updated 2 months ago
- Bijections & normalizing flows with JAX/NNX☆29Aug 23, 2026Updated last week