Prototype for sparse automatic differentiation in JAX + Julia
☆24Jun 17, 2025Updated last year
Alternatives and similar repositories for sparsediffax
Users that are interested in sparsediffax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- SParse lINear solvers in JAX☆55Aug 4, 2026Updated last month
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated 2 months ago
- Automatic Sparse Differentiation in JAX.☆52Sep 3, 2026Updated 3 weeks ago
- GPU-accelerated differentiable sparse linear solver library for JAX☆53Aug 30, 2026Updated 3 weeks ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆70Aug 1, 2026Updated last month
- 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.
- A complete example of batched refactorization in cuSOLVER.☆12Dec 22, 2021Updated 4 years ago
- JAXMg: A multi-GPU linear solver in JAX☆73Updated this week
- Multiple dispatch over abstract array types in JAX.☆143Updated this week
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆16May 20, 2024Updated 2 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆48Sep 18, 2026Updated last week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated 2 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆67Jan 23, 2026Updated 8 months ago
- Graphs defined by a grid of vertices. Compatible with the Graphs.jl ecosystem☆12Jun 22, 2026Updated 3 months ago
- ☆19Jul 25, 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.
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆14Jun 4, 2026Updated 3 months ago
- Matrix-free linear algebra in JAX.☆180Sep 2, 2026Updated 3 weeks ago
- A simple one file Python project to run 2D phase field problems with linear quadrilaterals☆22Sep 1, 2026Updated 3 weeks ago
- Generate Streamlit apps from Tesseracts, automagically.☆17Updated this week
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- ☆26Apr 27, 2026Updated 5 months ago
- Code to train a Multiple-Input Fourier Neural Operator (MIFNO) to predict the solution of 3D source-dependent Partial Differential Equati…☆15Jan 11, 2026Updated 8 months ago
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 11 months ago
- An unofficial Julia wrapper for the RAPIDS.ai ecosystem using PythonCall.jl☆23Apr 21, 2024Updated 2 years ago
- 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.
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆20Sep 18, 2026Updated last week
- Toolkit for Bayesian scaling analysis☆14Sep 8, 2022Updated 4 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆82Feb 12, 2026Updated 7 months ago
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Sep 17, 2026Updated last week
- Particle-mesh based calculations of long-range interactions in JAX☆32Updated this week
- Automated material constitutive modeling in JAX☆46Jul 23, 2026Updated 2 months ago
- Universal components for differentiable scientific computing.☆134Updated this week
- Differentiable Hydrodynamical Simulations☆18Updated this week
- ☆36Jul 19, 2023Updated 3 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.
- Coloring algorithms for sparse Jacobian and Hessian matrices☆45Sep 20, 2026Updated last week
- JAX translation of boltz☆32Aug 11, 2026Updated last month
- Calculation framework☆13Jan 20, 2026Updated 8 months ago
- ☆41Jun 29, 2026Updated 2 months ago
- Cut finite elements for FEniCSx☆20Jun 24, 2026Updated 3 months ago
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆62Updated this week
- Automatically exported from code.google.com/p/another-chebpy☆13Apr 28, 2015Updated 11 years ago