A JAX-native High Performance Eval Metrics Library
☆62Jul 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☆126Updated this week
- Minimal, lightweight JAX implementations of popular models.☆239May 29, 2026Updated last month
- ☆304Updated this week
- Library for reading and processing ML training data.☆756Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆526Updated 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.
- ☆21Jul 7, 2026Updated 2 weeks ago
- Coordinates in JAX☆35Updated this week
- Multi-backend recommender systems with Keras 3☆174Jul 16, 2026Updated last week
- Minimal yet performant LLM examples in pure JAX☆269Jul 4, 2026Updated 3 weeks ago
- Dynamical Systems with JAX☆12Jun 3, 2026Updated last month
- A Lightweight LLM Post-Training Library☆2,385Updated this week
- ☆19Jul 8, 2026Updated 2 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆64Jan 23, 2026Updated 6 months ago
- Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡☆32Updated this week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Reproduce analyses and baselines from the State paper☆16Jun 29, 2026Updated 3 weeks ago
- Special functions in Jax☆18Mar 1, 2025Updated last year
- Deep memory and sequence models in JAX☆32Jun 8, 2026Updated last month
- Stochastic trace estimation using JAX☆20Jun 29, 2026Updated 3 weeks ago
- ☆138Jul 7, 2026Updated 2 weeks ago
- OpenGraph image generation for crates.io packages☆15Jun 18, 2026Updated last month
- ☆377Updated this week
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆21Apr 20, 2023Updated 3 years ago
- Reinforcement learning with Equinox☆21Mar 4, 2025Updated last year
- 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.
- Evaluation suite for transcriptomic perturbation effect prediction models. Includes support for single-cell foundation models.☆39Jul 29, 2025Updated 11 months ago
- Investigating the role of pre-training dataset size and diversity on the performance of single-cell foundation models☆19May 12, 2026Updated 2 months ago
- a dataloader to work with large single cell datasets from lamindb☆41May 20, 2026Updated 2 months ago
- A simple, performant and scalable JAX-based world modeling codebase.☆155Jan 15, 2026Updated 6 months ago
- ☆26Jul 8, 2026Updated 2 weeks ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆19Jul 1, 2026Updated 3 weeks ago
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆193Updated this week
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆1,292Jul 13, 2026Updated last week
- Torch utilities for doing machine learning in gravitational wave physics☆34Updated 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.
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- ☆68Updated this week
- ☆67Aug 21, 2024Updated last year
- Fast and reliable distributed systems in Python☆35Mar 25, 2026Updated 4 months ago
- Bijections & normalizing flows with JAX/NNX☆23Jun 30, 2026Updated 3 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆135May 7, 2026Updated 2 months ago
- ☆17Nov 28, 2025Updated 7 months ago