A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
☆95Mar 31, 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A {crew} launcher plugin for AWS Batch☆18Jan 5, 2026Updated 3 months ago
- Additional censored event tools for Distributions.jl☆14Mar 31, 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 3 years ago
- State-space model inference with JAX☆55Mar 31, 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.
- A repository run by the Centers for Disease Control and Prevention (CDC) to collect forecasts of incident respiratory COVID-19 hospital a…☆24Apr 2, 2026Updated last week
- Active Statistics book web page☆12Jan 3, 2025Updated last year
- ☆13Dec 15, 2025Updated 3 months ago
- ☆27Jul 26, 2023Updated 2 years ago
- Bounding Interval Hierarchy and Bounding Volume Hierarchy for Node.js☆23Apr 15, 2018Updated 7 years ago
- ☆26Nov 8, 2023Updated 2 years ago
- ☆21Updated this week
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Apr 2, 2026Updated last week
- A comparison of PPL APIs☆27Jul 23, 2019Updated 6 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Apr 1, 2026Updated last week
- 基于 CUDA Driver API 的 cuda 运行时环境☆16Jul 30, 2025Updated 8 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
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆204Apr 1, 2026Updated last week
- ODE modelling in Python☆30Apr 1, 2026Updated last week
- ☆23Jun 6, 2025Updated 10 months ago
- Generic decision trees for rust☆12Sep 2, 2018Updated 7 years ago
- Package for data-driven and phenomenological gravitational waveform models☆12Mar 27, 2026Updated last week
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- Agentic Github Issues Retrieval on Kubernetes☆27Aug 5, 2025Updated 8 months ago
- 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☆21Mar 24, 2026Updated 2 weeks ago
- Variational Auto-Encoder based on Roberta encoder.☆12Oct 31, 2020Updated 5 years ago
- MLflow logging for PyMC☆14Aug 31, 2025Updated 7 months ago
- ✈️🏞️ Flying airplane in natural terrain generated by Perlin Noise.☆10Nov 20, 2022Updated 3 years ago
- open source smarthome radiator thermostat☆15Jan 3, 2024Updated 2 years ago
- ☆12Nov 6, 2024Updated last year
- Pytorch routines for (Ker)nel (Mac)hines☆11Oct 10, 2025Updated 5 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- 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 8 months ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated last year
- The jsii sample code transliterator☆29Updated this week
- Helper scripts I use to run many experiments in the morning to check at night☆20Jun 14, 2021Updated 4 years ago
- Slides and materials for workshop on "Two views on regression with PyMC3 and scikit-learn"☆18Apr 6, 2023Updated 3 years ago
- ☆34Feb 21, 2026Updated last month