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☆52Updated this week
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆21Jul 15, 2026Updated last week
- Automatic Sparse Differentiation in JAX.☆37Updated this week
- GPU-accelerated differentiable sparse linear solver library for JAX☆35Updated this week
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆70Jul 1, 2026Updated 3 weeks ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A complete example of batched refactorization in cuSOLVER.☆13Dec 22, 2021Updated 4 years ago
- JAXMg: A multi-GPU linear solver in JAX☆58Updated this week
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆142Updated this week
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆17May 20, 2024Updated 2 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆47Apr 15, 2026Updated 3 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆47Jul 10, 2026Updated 2 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆64Jan 23, 2026Updated 6 months ago
- Graphs defined by a grid of vertices. Compatible with the Graphs.jl ecosystem☆12Jun 22, 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.
- ☆19Updated this week
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆14Jun 4, 2026Updated last month
- Matrix-free linear algebra in JAX.☆178Jul 1, 2026Updated 3 weeks ago
- A simple one file Python project to run 2D phase field problems with linear quadrilaterals☆20Aug 25, 2025Updated 11 months ago
- Generate Streamlit apps from Tesseracts, automagically.☆16Updated this week
- flexible meta-learning in jax☆16Oct 19, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- ☆25Apr 27, 2026Updated 3 months ago
- Simple Xarray + JAX Integration☆21Mar 21, 2026Updated 4 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A simulator for ecological dynamics written in Julia.☆27Updated this week
- Code to train a Multiple-Input Fourier Neural Operator (MIFNO) to predict the solution of 3D source-dependent Partial Differential Equati…☆14Jan 11, 2026Updated 6 months ago
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 9 months ago
- Parallel solution of structured nonlinear optimization problems☆13Dec 16, 2025Updated 7 months ago
- StateSpaceLearning.jl is a Julia package for time-series analysis using state space learning framework.☆21May 31, 2026Updated last month
- An unofficial Julia wrapper for the RAPIDS.ai ecosystem using PythonCall.jl☆22Apr 21, 2024Updated 2 years ago
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆20Jun 24, 2026Updated last month
- Toolkit for Bayesian scaling analysis☆14Sep 8, 2022Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆82Feb 12, 2026Updated 5 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Jul 3, 2026Updated 3 weeks ago
- Particle-mesh based calculations of long-range interactions in JAX☆30Jul 22, 2026Updated last week
- Universal components for differentiable scientific computing.☆117Updated this week
- Automated material constitutive modeling in JAX☆44Updated this week
- Differentiable Hydrodynamical Simulations☆17Jul 17, 2026Updated last week
- Mathematical Programming in JAX☆166Jun 2, 2025Updated last year
- ☆36Jul 19, 2023Updated 3 years ago