A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
☆180Jul 3, 2026Updated this 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:
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆16May 28, 2025Updated last year
- Code for "What really matters in matrix-whitening optimizers?"☆24Oct 31, 2025Updated 8 months ago
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆16Aug 6, 2024Updated last year
- Jaxpr Visualisation Tool☆37Dec 22, 2024Updated last year
- ☆13Jun 22, 2026Updated last week
- 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.
- Disciplined Convex Programming in R using Convex.jl.☆14Dec 18, 2018Updated 7 years ago
- An adiabatic waveform generator using black hole perturbation theory☆20Oct 17, 2024Updated last year
- US RSV Forecast Hub☆12Jun 5, 2026Updated 3 weeks ago
- A {crew} launcher plugin for AWS Batch☆19Jan 5, 2026Updated 5 months ago
- Python package for calculating pulsar timing array sensitivity curves.☆17Mar 20, 2026Updated 3 months ago
- Additional censored event tools for Distributions.jl☆14Updated this week
- Companion tutorials for the NANOGrav 15-year GWB analysis☆20Aug 11, 2025Updated 10 months ago
- ☆14Jan 27, 2022Updated 4 years ago
- AI Skills for Antithesis Users☆69Jun 24, 2026Updated last week
- 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.
- VeLO optimizer in PyTorch☆20Feb 6, 2023Updated 3 years ago
- Differentiable interface to Firedrake for JAX☆15Feb 28, 2021Updated 5 years ago
- State-space model inference with JAX☆71Updated this week
- Provides filesystem access for the Rhai scripting language.☆15Jun 10, 2026Updated 3 weeks ago
- SeismicNet: Physics-informed neural networks for seismic wave modeling in semi-infinite domain☆22May 5, 2024Updated 2 years ago
- ArviZ modular plotting☆20Updated this week
- A repository run by the Centers for Disease Control and Prevention (CDC) to collect forecasts of incident respiratory COVID-19 hospital a…☆27Updated this week
- An easy to use interface to gravitational wave surrogate models☆39Jun 19, 2026Updated 2 weeks ago
- Full Bayesian inference for neural networks using TensorFlow☆15Aug 19, 2021Updated 4 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆15Jun 8, 2026Updated 3 weeks ago
- Analysis and tutorial materials associated with the NANOGrav Collaboration's search for an isotropic stochastic gravitational-wave backgr…☆32Oct 22, 2023Updated 2 years ago
- Bounding Interval Hierarchy and Bounding Volume Hierarchy for Node.js☆23Apr 15, 2018Updated 8 years ago
- A file server☆37Dec 6, 2025Updated 6 months ago
- AI-assisted workflows that follow Very Good Ventures best practices. It pairs perfectly with our Very Good AI Flutter Plugin.☆79Jun 19, 2026Updated 2 weeks ago
- ☆27Nov 8, 2023Updated 2 years ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆16Updated this week
- ☆34Updated this week
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Jun 22, 2026Updated last week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Rickbot ADK - a multi-personality chatbot built using Google ADK and Gemini, and leveraging Agent-Starter-Pack and Gemini CLI☆19Updated this week
- ☆13May 14, 2025Updated last year
- ☆33Updated this week
- ☆13Mar 26, 2026Updated 3 months ago
- Gravitational-wave astronomy in JAX☆15Updated this week
- Automatic Differentiation for R☆23Dec 18, 2018Updated 7 years ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago