A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
☆115Apr 22, 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:
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆15May 28, 2025Updated 11 months ago
- Code for "What really matters in matrix-whitening optimizers?"☆23Oct 31, 2025Updated 5 months ago
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆17Aug 6, 2024Updated last year
- Data and statistical models for biomarker shedding.☆16Apr 2, 2026Updated 3 weeks ago
- ☆13Apr 13, 2026Updated 2 weeks ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Codebase for EnterpriseOps-Gym from ServiceNow☆82Mar 25, 2026Updated last month
- US RSV Forecast Hub☆12Updated this week
- A {crew} launcher plugin for AWS Batch☆18Jan 5, 2026Updated 3 months ago
- Additional censored event tools for Distributions.jl☆14Updated this week
- 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 last year
- Package for probabilistic forecast reconciliation via score optimisation☆13Apr 5, 2023Updated 3 years ago
- Differentiable interface to Firedrake for JAX☆15Feb 28, 2021Updated 5 years ago
- Provides filesystem access for the Rhai scripting language.☆14Oct 25, 2023Updated 2 years ago
- 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.
- State of the art inference for your bayesian models.☆239Jan 14, 2026Updated 3 months ago
- Cache Commander — a TUI and MCP server to explore, audit, and clean developer cache directories. Scan for CVEs, find outdated packages, r…☆64Apr 21, 2026Updated last week
- A modular, HTTP-driven orchestrator to define, practice, and refine pipelines and generative tools☆19Apr 27, 2025Updated last year
- A repository run by the Centers for Disease Control and Prevention (CDC) to collect forecasts of incident respiratory COVID-19 hospital a…☆24Updated this week
- Active Statistics book web page☆12Jan 3, 2025Updated last year
- ☆13Dec 15, 2025Updated 4 months ago
- A design approach to implementing a good GitHub Wiki system of documentation. This repository explains how to create a GitHub Wiki and th…☆18Mar 19, 2025Updated last year
- A GPU-powered Interactive Graphics Device for R☆28Dec 2, 2024Updated last year
- A file server☆36Dec 6, 2025Updated 4 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Train YOLO + VLM with one command. Auto-generate vision-language training data from YOLO labels - no extra labeling needed.☆30Apr 21, 2026Updated last week
- Bounding Interval Hierarchy and Bounding Volume Hierarchy for Node.js☆23Apr 15, 2018Updated 8 years ago
- This GenAI solution enables users to extract insights from diverse data formats (video, audio, PDFs, text) through a unified interface. U…☆19Feb 12, 2026Updated 2 months ago
- ☆21Updated this week
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Apr 20, 2026Updated last week
- ☆13Mar 26, 2026Updated last month
- ☆13May 14, 2025Updated 11 months ago
- A command-line tool to enrich your database schema with metadata (primarily column comments) and easily manage those comments. It's desig…☆26Apr 21, 2026Updated last week
- 基于 CUDA Driver API 的 cuda 运行时环境☆16Jul 30, 2025Updated 9 months ago
- AI Agents on 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 Towers of Hanoi environment in OpenAI Gym Style☆14Jun 6, 2019Updated 6 years ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 2 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…☆216Apr 23, 2026Updated last week
- Cyber Pilot is a traceable delivery system for requirements, design, plans, and code.☆55Apr 14, 2026Updated 2 weeks ago
- GitHub tools and agents for the Vercel AI SDK, wrap GitHub's API as AI SDK tools ready to plug into any agent☆85Updated this week
- Generic decision trees for rust☆12Sep 2, 2018Updated 7 years ago
- TypeScript repo for the XDK auto-generated code.☆55Feb 28, 2026Updated 2 months ago