A JAX-native High Performance Eval Metrics Library
☆60Jun 3, 2026Updated last 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☆120Jun 8, 2026Updated last week
- Minimal, lightweight JAX implementations of popular models.☆236May 29, 2026Updated 2 weeks ago
- Minimal yet performant LLM examples in pure JAX☆256Apr 10, 2026Updated 2 months ago
- Multi-backend recommender systems with Keras 3☆173Jun 1, 2026Updated 2 weeks ago
- Tokamax: A GPU and TPU kernel library.☆227Updated this week
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- A Lightweight LLM Post-Training Library☆2,341Updated this week
- Create a <style> element with CSS properties, filtering input using an allowlist or blocklist.☆18Updated this week
- Experiments Notebook of "Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism"☆15Apr 30, 2025Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆18Aug 30, 2025Updated 9 months ago
- Feature engineering for big data and quick inference☆21Updated this week
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- Dynamical Systems with JAX☆12Jun 3, 2026Updated last week
- Codes to reproduce results in the paper☆10Apr 9, 2021Updated 5 years 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.
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 3 months ago
- LLM extensions for Sphinx Documentation☆30Apr 2, 2026Updated 2 months ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆15May 25, 2026Updated 3 weeks ago
- Vecchia approximations for Gaussian log-likelihoods☆15Apr 27, 2026Updated last month
- A script for getting a BibTeX entry from a URL, DOI, or arXiv ID☆10Jun 7, 2026Updated last week
- ☆21May 9, 2026Updated last month
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆14Updated this week
- Gravitational-wave astronomy in JAX☆14Jun 8, 2026Updated last week
- KENN - A Transformer Encoder for Gravitational Waves☆12Sep 29, 2025Updated 8 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆17Apr 7, 2025Updated last year
- Automated discovery of optimal branching rules for the branch-and-bound algorithm☆19Dec 15, 2025Updated 6 months ago
- Polynomials for efficiently computing the density of seawater☆15May 27, 2026Updated 2 weeks ago
- A lightweight code to generate mock gravitational wave detector data using user-defined population and arbritary detector sensitivity.☆15Apr 29, 2026Updated last month
- Package for data-driven and phenomenological gravitational waveform models☆12Updated this week
- Superphot+, upgraded version of Superphot that uses nested sampling☆12Dec 31, 2025Updated 5 months ago
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 10 months ago
- YetAnotherWandbClient☆13Mar 16, 2026Updated 2 months ago
- ☆16Updated this week
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Framework for computing kilonova light curves☆11Sep 2, 2025Updated 9 months ago
- Perturbational analysis by causality-aware generative model for single-cell RNA-sequencing data☆23Dec 16, 2025Updated 5 months ago
- Minibatch loading for on-disk anndata files, co-developed by Lamin Labs and scverse☆62Updated this week
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆60Jan 23, 2026Updated 4 months ago
- 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