Automatic Sparse Differentiation in JAX.
☆59Oct 21, 2026Updated last week
Alternatives and similar repositories for asdex
Users that are interested in asdex are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- SParse lINear solvers in JAX☆59Aug 4, 2026Updated 2 months ago
- GPU-accelerated differentiable sparse linear solver library for JAX☆58Updated this week
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆71Aug 1, 2026Updated 2 months ago
- Algebraic multigrid solvers in JAX☆43Aug 27, 2026Updated 2 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.
- Universal PET models in JAX☆12Updated this week
- Coloring algorithms for sparse Jacobian and Hessian matrices☆46Updated this week
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated 3 months ago
- Differentiable ensemble solvers for differential equations on GPUs, in JAX☆40Sep 3, 2026Updated last month
- ☆34Oct 23, 2026Updated last week
- tools for graph-based machine-learning potentials in jax☆28Apr 9, 2024Updated 2 years ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆33Updated this week
- JAXMg: A multi-GPU linear solver in JAX☆113Oct 22, 2026Updated last week
- Matrix-free linear algebra in JAX.☆183Oct 22, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Euclid-equivariant operations for JAX☆27Oct 15, 2026Updated 2 weeks ago
- Lego-like building blocks for Finite Element Analysis☆70Oct 19, 2026Updated last week
- Automated material constitutive modeling in JAX☆48Oct 23, 2026Updated last week
- Solve sparse linear systems in JAX using the KLU algorithm☆48Sep 18, 2026Updated last month
- A benchmark suite and re-usable collection of differentiable physics solvers.☆27Updated this week
- Particle-mesh based calculations of long-range interactions in JAX☆32Oct 20, 2026Updated last week
- ☆37Jul 19, 2023Updated 3 years ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 4 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆587Sep 26, 2026Updated last month
- 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.
- A simulation node in your JAX workflow.☆90Oct 23, 2026Updated last week
- Website for the book "The Elements of Differentiable Programming".☆15Updated this week
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆80Apr 5, 2026Updated 6 months ago
- Adjoint framework for DOLFINx☆17Updated this week
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆15Oct 23, 2026Updated last week
- Ultra fast atomistic storage database for random reads and ML training☆13Sep 30, 2026Updated last month
- General-purpose differentiable meshfree solver built on JAX and the Material Point Method (MPM), supporting GPU-accelerated forward and i…☆17Jun 12, 2026Updated 4 months ago
- Fast operator-overloading Jacobian & Hessian sparsity pattern detection.☆61Oct 7, 2026Updated 3 weeks ago
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆63Sep 26, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆14Jul 2, 2023Updated 3 years ago
- Graph Neural Networks with JAX/NNX☆20Aug 12, 2026Updated 2 months ago
- Official repository for PosteriorBench: From Point Estimates to Posterior Matching in Evaluating Generative Solvers for Spatial Field Inv…☆16Oct 9, 2026Updated 3 weeks ago
- differentiable (magneto)hydrodynamics for astrophysics in JAX☆74Oct 20, 2026Updated last week
- An interface to various automatic differentiation backends in Julia.☆316Oct 19, 2026Updated last week
- A Differentiable Topology Optimization Framework for Multiphysics Fluidic Problems☆28Mar 6, 2026Updated 7 months ago
- Repository for automatic differentiation backend types☆73Oct 17, 2026Updated last week