A JAX-native High Performance Eval Metrics Library
☆58Feb 3, 2026Updated last month
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☆102Updated this week
- Minimal, lightweight JAX implementations of popular models.☆214Mar 4, 2026Updated 3 weeks ago
- Minimal yet performant LLM examples in pure JAX☆245Jan 14, 2026Updated 2 months ago
- Multi-backend recommender systems with Keras 3☆163Mar 3, 2026Updated 3 weeks ago
- Tokamax: A GPU and TPU kernel library.☆185Mar 19, 2026Updated last week
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- MOSEK Optimizer API for Rust☆16Feb 19, 2026Updated last month
- Experiments Notebook of "Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism"☆15Apr 30, 2025Updated 10 months ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆16Aug 30, 2025Updated 6 months ago
- Demos and benchmarks for TopOpt.jl @WCSMO-21☆11Dec 2, 2021Updated 4 years ago
- Dynamical Systems with JAX☆12Jan 11, 2026Updated 2 months ago
- Codes to reproduce results in the paper☆10Apr 9, 2021Updated 4 years ago
- A machine learning framework with readable source code☆14Apr 30, 2025Updated 10 months ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Mar 13, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- A script for getting a BibTeX entry from a URL, DOI, or arXiv ID☆10Oct 23, 2024Updated last year
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆12Updated this week
- A library for computing the singular value decomposition of large and sparse or structured matrices.☆14Mar 16, 2026Updated last week
- KENN - A Transformer Encoder for Gravitational Waves☆11Sep 29, 2025Updated 5 months ago
- Rescure useful code in TensorNetwork by google for further personal usage.☆13Dec 6, 2024Updated last year
- Minibatch loading for on-disk anndata files, co-developed by Lamin Labs and scverse☆27Updated this week
- A lightweight code to generate mock gravitational wave detector data using user-defined population and arbritary detector sensitivity.☆15Mar 6, 2026Updated 2 weeks ago
- Package for data-driven and phenomenological gravitational waveform models☆12Feb 12, 2025Updated last year
- Superphot+, upgraded version of Superphot that uses nested sampling☆12Dec 31, 2025Updated 2 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated last year
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 7 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆53Jan 23, 2026Updated 2 months ago
- YetAnotherWandbClient☆13Mar 16, 2026Updated last week
- ☆13Aug 27, 2025Updated 6 months ago
- Large scale convex optimization solvers in julia☆23Jan 4, 2021Updated 5 years ago
- ☆15Dec 11, 2025Updated 3 months ago
- Framework for computing kilonova light curves☆12Sep 2, 2025Updated 6 months ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆10Oct 23, 2024Updated last year
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Experiments to run single cell analyses efficiently at scale using Zarr, anndata, Scanpy, and Apache Spark☆15Mar 28, 2019Updated 6 years ago
- Easy interoperability with Automatic Differentiation libraries through NumPy interface to Firedrake and FEniCS☆15Nov 12, 2023Updated 2 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
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆18Apr 20, 2023Updated 2 years ago