A JAX-native High Performance Eval Metrics Library
☆64Aug 11, 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☆128Updated this week
- Minimal, lightweight JAX implementations of popular models.☆241May 29, 2026Updated 2 months ago
- ☆310Updated this week
- Library for reading and processing ML training data.☆766Updated this week
- ☆21Jul 7, 2026Updated last month
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated last year
- Coordinates in JAX☆37Updated this week
- Multi-backend recommender systems with Keras 3☆175Aug 1, 2026Updated 2 weeks ago
- Minimal yet performant LLM examples in pure JAX☆272Jul 4, 2026Updated last month
- Dynamical Systems with JAX☆12Jun 3, 2026Updated 2 months ago
- A Lightweight LLM Post-Training Library☆2,405Updated this week
- ☆19Jul 8, 2026Updated last month
- Performing parameter estimation on gravitational wave data with machine learning☆21May 4, 2026Updated 3 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆65Jan 23, 2026Updated 6 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.
- Code for the ICML 2023 paper "Geometric Autoencoders - What You See is What You Decode"☆17Jul 28, 2023Updated 3 years ago
- Special functions in Jax☆18Mar 1, 2025Updated last year
- JAX-Toolbox☆426Updated this week
- Stochastic trace estimation using JAX☆20Jun 29, 2026Updated last month
- ☆379Updated this week
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆21Apr 20, 2023Updated 3 years ago
- Target Policy Optimization (JAX)☆31Apr 18, 2026Updated 3 months ago
- Evaluation suite for transcriptomic perturbation effect prediction models. Includes support for single-cell foundation models.☆39Jul 29, 2025Updated last year
- Generative Models for Simulation-Based Inference in JAX☆19Jul 23, 2026Updated 3 weeks 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.
- Investigating the role of pre-training dataset size and diversity on the performance of single-cell foundation models☆20May 12, 2026Updated 3 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.☆159Jan 15, 2026Updated 7 months ago
- pytorch dataloaders for single-cell perturbation data☆50Jul 18, 2026Updated 3 weeks ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 7 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆19Updated this week
- ROUGE L metric implementation using tensorflow ops☆12Sep 17, 2018Updated 7 years ago
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆193Aug 4, 2026Updated last week
- ☆37Oct 31, 2025Updated 9 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.
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆1,330Jul 27, 2026Updated 2 weeks ago
- 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
- ☆17May 20, 2026Updated 2 months ago
- ☆67Aug 6, 2026Updated last week
- ☆67Aug 21, 2024Updated last year
- Fast and reliable distributed systems in Python☆35Mar 25, 2026Updated 4 months ago