A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
☆70Mar 13, 2026Updated last week
Alternatives and similar repositories for jax-mps
Users that are interested in jax-mps are comparing it to the libraries listed below
Sorting:
- Data and statistical models for biomarker shedding.☆15Feb 23, 2026Updated 3 weeks ago
- ☆13Mar 9, 2026Updated last week
- A {crew} launcher plugin for AWS Batch☆18Jan 5, 2026Updated 2 months ago
- Additional censored event tools for Distributions.jl☆14Mar 10, 2026Updated last week
- Models and infrastructure for forecasting respiratory disease hospitalizations using wastewater data with PyRenew.☆16Updated this week
- Package for probabilistic forecast reconciliation via score optimisation☆13Apr 5, 2023Updated 2 years ago
- Active Statistics book web page☆12Jan 3, 2025Updated last year
- ☆13Dec 15, 2025Updated 3 months ago
- A GPU-powered Interactive Graphics Device for R☆28Dec 2, 2024Updated last year
- Bounding Interval Hierarchy and Bounding Volume Hierarchy for Node.js☆23Apr 15, 2018Updated 7 years ago
- A comparison of PPL APIs☆27Jul 23, 2019Updated 6 years ago
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Mar 9, 2026Updated last week
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆196Updated this week
- 基于 CUDA Driver API 的 cuda 运行时环境☆14Jul 30, 2025Updated 7 months ago
- A repository to store COVID-19 variant nowcasts collected as a modeling hub.☆21Updated this week
- A Towers of Hanoi environment in OpenAI Gym Style☆14Jun 6, 2019Updated 6 years ago
- ODE modelling in Python☆30Feb 25, 2026Updated 3 weeks ago
- Parametric differentiable curves with PyTorch for continuous embeddings, shape-restricted models, or KANs☆53Feb 17, 2026Updated last month
- My solutions to the projects set in CMU's Intro to Database Systems Course in Rust☆10Jul 14, 2023Updated 2 years ago
- Waveform model for inspiralling precessing-eccentric compact binaries☆19Oct 30, 2025Updated 4 months ago
- ☆16May 29, 2025Updated 9 months ago
- A implementation of NUTS in rust☆97Updated this week
- ✈️🏞️ Flying airplane in natural terrain generated by Perlin Noise.☆10Nov 20, 2022Updated 3 years ago
- MLflow logging for PyMC☆14Aug 31, 2025Updated 6 months ago
- ☆12Nov 6, 2024Updated last year
- A benchmark of meaningful graph datasets with tabular node features☆14Oct 29, 2025Updated 4 months ago
- Collaborative inference of latent diffusion via hivemind☆12May 29, 2023Updated 2 years ago
- Code repository for CIERA's Research Experiences in Astronomy at CIERA for High School Students (REACH) Program☆14Jul 22, 2025Updated 7 months ago
- This repository contains notebooks with different probability density function estimators.☆14Jun 4, 2020Updated 5 years ago
- Predict performance of a centrifugal chiller using Multiple Linear Regression☆11May 17, 2015Updated 10 years ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated last year
- crew launcher plugins for traditional high-performance computing clusters☆33Sep 15, 2025Updated 6 months ago
- Slides and materials for workshop on "Two views on regression with PyMC3 and scikit-learn"☆18Apr 6, 2023Updated 2 years ago
- Python implementation of plot from Kay, Kola, Hullman, Munson "When (ish) is My Bus?" (2016)☆18Dec 19, 2019Updated 6 years ago
- Testing the consistency of binary classification performance scores reported in papers☆12Aug 21, 2025Updated 6 months ago
- Emacs support for Jupyter notebooks☆15Nov 17, 2024Updated last year
- Mastering Gaussian Processes with PyMC☆18Aug 18, 2025Updated 7 months ago
- A Probabilistic Programming Language In Rust☆41Jun 25, 2022Updated 3 years ago
- Clean version of research-projects, just with ILE☆18Mar 10, 2026Updated last week