A JAX-native High Performance Eval Metrics Library
☆64Sep 22, 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☆136Updated this week
- Minimal, lightweight JAX implementations of popular models.☆245May 29, 2026Updated 3 months ago
- ☆311Updated this week
- Library for reading and processing ML training data.☆782Updated this week
- ☆21Sep 14, 2026Updated last week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated last year
- Coordinates in JAX☆38Updated this week
- Minimal yet performant LLM examples in pure JAX☆284Aug 23, 2026Updated last month
- Tokamax: A GPU and TPU kernel library.☆303Updated this week
- Dynamical Systems with JAX☆12Jun 3, 2026Updated 3 months ago
- A Lightweight LLM Post-Training Library☆2,460Updated this week
- ☆19Sep 2, 2026Updated 3 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆67Jan 23, 2026Updated 8 months ago
- Code for the ICML 2023 paper "Geometric Autoencoders - What You See is What You Decode"☆17Jul 28, 2023Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- 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
- Deep memory and sequence models in JAX☆33Jun 8, 2026Updated 3 months ago
- ☆389Updated this week
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆22Apr 20, 2023Updated 3 years ago
- Reinforcement learning with Equinox☆21Mar 4, 2025Updated last year
- Target Policy Optimization (JAX)☆31Apr 18, 2026Updated 5 months ago
- Evaluation suite for transcriptomic perturbation effect prediction models. Includes support for single-cell foundation models.☆41Jul 29, 2025Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆164Jan 15, 2026Updated 8 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- pytorch dataloaders for single-cell perturbation data☆54Jul 18, 2026Updated 2 months ago
- 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.☆23Sep 10, 2026Updated 2 weeks ago
- ☆38Oct 31, 2025Updated 10 months ago
- Code accompanying the NeurIPS 2019 paper AutoAssist: A Framework to Accelerate Training of Deep Neural Networks.☆14Oct 3, 2022Updated 3 years ago
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆1,449Updated this week
- Torch utilities for doing machine learning in gravitational wave physics☆34Updated this week
- This is the official repository for the ICLR 2026 paper "The Curious Case of In-Training Compression of State Space Models".☆34Feb 23, 2026Updated 7 months ago
- 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.
- Fast and reliable distributed systems in Python☆36Mar 25, 2026Updated 5 months ago
- A Python library for defining structured command-line flags.☆35Jun 10, 2026Updated 3 months ago
- A test app to determine what Project Fugu 🐡 APIs your browser supports.☆21Jun 14, 2026Updated 3 months ago
- Bijections & normalizing flows with JAX/NNX☆30Aug 23, 2026Updated last month
- A State-Space Model with Rational Transfer Function Representation.☆85May 17, 2024Updated 2 years ago
- Neural Emulator Architectures in JAX.☆27Aug 25, 2026Updated last month
- Bringing StableHLO to Apple Silicon☆24Jun 6, 2026Updated 3 months ago