Algebraic multigrid solvers in JAX
☆43Aug 27, 2026Updated 2 months ago
Alternatives and similar repositories for AMJax
Users that are interested in AMJax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- GPU-accelerated differentiable sparse linear solver library for JAX☆58Oct 26, 2026Updated last week
- SParse lINear solvers in JAX☆59Aug 4, 2026Updated 3 months ago
- JAXMg: A multi-GPU linear solver in JAX☆113Updated this week
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- An Automized Partial Differential Equation solver based on JAX☆14Jul 9, 2026Updated 4 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Adjoint framework for DOLFINx☆18Updated this week
- A simulation node in your JAX workflow.☆91Updated this week
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆15Updated this week
- Training methodologies for autoregressive neural operators/emulators in JAX.☆16Jun 1, 2026Updated 5 months ago
- A JAX-based adaptive mesh refinement framework☆36Jul 14, 2026Updated 3 months ago
- Efficient GPU-computing simulation platform for differentiable phase field model☆28Apr 14, 2026Updated 6 months ago
- Neural Emulator Architectures in JAX.☆29Aug 25, 2026Updated 2 months ago
- A benchmark suite for differentiable physics solvers.☆27Updated this week
- Minimalist block diagram GUI for control systems☆13Feb 1, 2026Updated 9 months 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.
- Automated material constitutive modeling in JAX☆48Oct 28, 2026Updated last week
- ☆15Updated this week
- torch sparse linear algebra library☆58Oct 19, 2026Updated 3 weeks ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆33Updated this week
- Matrix-free linear algebra in JAX.☆183Updated this week
- Taylor mode automatic differentiation (jets) in PyTorch☆15Updated this week
- Repository for “Introduction to Generative Art and Scientific Visualization” mini-course taught at Harvard January@GSAS☆16Jan 15, 2023Updated 3 years ago
- ☆12May 9, 2026Updated 6 months ago
- Machine Learning-Enabled Compact Photonic Tensor Core based on Programmable Multi-Operand Multimode Interference☆14Sep 23, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Differentiable implicit CAD in the browser: edit via GUI or code, solve constraints, render with WebGPU raymarching, optimize shapes end-…☆71Oct 23, 2026Updated 2 weeks ago
- Differentiable simulation of proliferating and active cell clusters.☆15Nov 1, 2026Updated last week
- A symbolic finite element definition library☆73Oct 14, 2026Updated 3 weeks ago
- helmholtz-x☆10Feb 10, 2025Updated last year
- jaxhps: an elliptic PDE solver built with machine learning in mind☆14Apr 7, 2026Updated 7 months ago
- Solvers for Implicit Runge Kutta methods☆35Updated this week
- Differentiable ensemble solvers for differential equations on GPUs, in JAX☆40Sep 3, 2026Updated 2 months ago
- Differentiable interface to FEniCS for JAX☆60May 30, 2021Updated 5 years ago
- Differentiable Fluid Dynamics Package☆666Aug 28, 2026Updated 2 months 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.
- ☆36Jul 11, 2017Updated 9 years ago
- An Open-Source Framework for Advanced Phase-Field Simulations☆79Oct 15, 2026Updated 3 weeks ago
- Spatial Modeling Algorithms for Reactions and Transport☆14Jun 24, 2026Updated 4 months ago
- ☆16Updated this week
- Dasslc is a solver for differential-algebraic equations inspired by DASSL. A python module for dasslc is provided.☆18Mar 7, 2024Updated 2 years ago
- Universal components for differentiable scientific computing.☆138Updated this week
- Graphs defined by a grid of vertices. Compatible with the Graphs.jl ecosystem☆12Oct 28, 2026Updated last week