A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
☆151Jun 5, 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:
- Multi-agent orchestration framework for AI applications - build, deploy, and manage AI agents across the full lifecycle with Forge, Conve…☆32Mar 28, 2026Updated 2 months ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆16May 28, 2025Updated last year
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆16Aug 6, 2024Updated last year
- Data and statistical models for biomarker shedding.☆17May 18, 2026Updated 3 weeks ago
- An experimental IDE for solid modeling with distance fields☆29May 13, 2026Updated last month
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Jaxpr Visualisation Tool☆37Dec 22, 2024Updated last year
- ☆13Jun 1, 2026Updated last week
- Codebase for EnterpriseOps-Gym from ServiceNow☆96Jun 3, 2026Updated last week
- Disciplined Convex Programming in R using Convex.jl.☆14Dec 18, 2018Updated 7 years ago
- US RSV Forecast Hub☆12Jun 5, 2026Updated last week
- Fine-Grained Distributed Computing☆11Feb 15, 2016Updated 10 years ago
- ☆12Dec 29, 2018Updated 7 years ago
- Models and infrastructure for routine forecasting of respiratory disease burden☆16Updated this week
- Plotutils+ is a fork of the GNU plotutils package, with fixes and extensions☆13May 14, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- VeLO optimizer in PyTorch☆20Feb 6, 2023Updated 3 years ago
- State-space model inference with JAX☆66Jun 4, 2026Updated last week
- Tools and prompt templates used to build and evaluate SWE-rebench-v2 tasks for the paper.☆62Mar 12, 2026Updated 3 months ago
- State of the art inference for your bayesian models.☆242Jan 14, 2026Updated 4 months ago
- SeismicNet: Physics-informed neural networks for seismic wave modeling in semi-infinite domain☆22May 5, 2024Updated 2 years ago
- Build autonomous Python agents with native Agent-to-Agent (A2A) communication☆25Updated this week
- A project between Anyscale and deepsense.ai implementing a cross-modal search application for e-commerce☆13Jun 5, 2024Updated 2 years ago
- Cache Commander — a TUI and MCP server to explore, audit, and clean developer cache directories. Scan for CVEs, find outdated packages, r…☆65May 15, 2026Updated 3 weeks ago
- Full Bayesian inference for neural networks using TensorFlow☆15Aug 19, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Quasi-steady electromagnetics including filamentized approximations, Biot-Savart, and Grad-Shafranov.☆13Jun 5, 2026Updated last week
- Active Statistics book web page☆12Jan 3, 2025Updated last year
- ☆14Dec 15, 2025Updated 5 months ago
- ☆27Jul 26, 2023Updated 2 years ago
- Simple process engineering calculation with Python☆23Jul 30, 2020Updated 5 years ago
- Integrate Amazon Q Business APIs into custom applications using Identity-aware credentials.☆13Nov 14, 2024Updated last year
- This GenAI solution enables users to extract insights from diverse data formats (video, audio, PDFs, text) through a unified interface. U…☆21Feb 12, 2026Updated 4 months ago
- ☆32Updated this week
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Apr 20, 2026Updated last month
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Gravitational-wave astronomy in JAX☆14Jun 5, 2026Updated last week
- 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 7 years ago
- Automatic Differentiation for R☆23Dec 18, 2018Updated 7 years ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 2 years ago
- ☆29Nov 12, 2025Updated 7 months ago
- ODE modelling in Python☆30Updated this week