A JAX-native High Performance Eval Metrics Library
☆56Apr 10, 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:
- Minimal yet performant LLM examples in pure JAX☆246Updated this week
- Multi-backend recommender systems with Keras 3☆164Updated this week
- Tokamax: A GPU and TPU kernel library.☆198Updated this week
- A Lightweight LLM Post-Training Library☆2,215Updated this week
- MOSEK Optimizer API for Rust☆16Apr 7, 2026Updated last week
- Deploy open-source AI quickly and easily - Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated last year
- Demos and benchmarks for TopOpt.jl @WCSMO-21☆11Dec 2, 2021Updated 4 years ago
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- Dynamical Systems with JAX☆12Jan 11, 2026Updated 3 months ago
- A machine learning framework with readable source code☆14Apr 30, 2025Updated 11 months ago
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated last month
- Vecchia approximations for Gaussian log-likelihoods☆14Apr 2, 2026Updated last week
- A script for getting a BibTeX entry from a URL, DOI, or arXiv ID☆10Oct 23, 2024Updated last year
- ☆21Mar 18, 2026Updated 3 weeks ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- A library for computing the singular value decomposition of large and sparse or structured matrices.☆14Mar 16, 2026Updated last month
- KENN - A Transformer Encoder for Gravitational Waves☆12Sep 29, 2025Updated 6 months ago
- A lightweight code to generate mock gravitational wave detector data using user-defined population and arbritary detector sensitivity.☆15Mar 6, 2026Updated last month
- Package for data-driven and phenomenological gravitational waveform models☆12Mar 27, 2026Updated 2 weeks ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated last year
- Superphot+, upgraded version of Superphot that uses nested sampling☆12Dec 31, 2025Updated 3 months ago
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 8 months ago
- A fast, differentiable, and extensible public BBN code☆16Mar 6, 2026Updated last month
- ☆13Aug 27, 2025Updated 7 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.
- YetAnotherWandbClient☆13Mar 16, 2026Updated 3 weeks ago
- ☆15Dec 11, 2025Updated 4 months ago
- Framework for computing kilonova light curves☆11Sep 2, 2025Updated 7 months ago
- Perturbational analysis by causality-aware generative model for single-cell RNA-sequencing data☆21Dec 16, 2025Updated 4 months ago
- ☆10Oct 23, 2024Updated last year
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago
- Experiments to run single cell analyses efficiently at scale using Zarr, anndata, Scanpy, and Apache Spark☆15Mar 28, 2019Updated 7 years ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆56Jan 23, 2026Updated 2 months ago
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- API for generating wallet passes from EU COVID-19 Vaccination Certificates☆12Jun 28, 2021Updated 4 years ago
- Easy interoperability with Automatic Differentiation libraries through NumPy interface to Firedrake and FEniCS☆15Nov 12, 2023Updated 2 years ago
- A project for implementing ML and NLP papers☆13May 22, 2020Updated 5 years ago
- A simple, performant and scalable JAX-based world modeling codebase.☆143Jan 15, 2026Updated 3 months ago
- Standalone minimal REPL + viewer for ao-guile [DEPRECATED]☆17Dec 24, 2017Updated 8 years ago
- OpenVDB fluid simulation using Material Point Method☆15Jun 14, 2018Updated 7 years ago
- Reproduce analyses and baselines from the State paper☆14Feb 25, 2026Updated last month