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☆59Aug 4, 2026Updated 3 months ago
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated 3 months ago
- Automatic Sparse Differentiation in JAX.☆68Oct 21, 2026Updated 2 weeks ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆71Aug 1, 2026Updated 3 months ago
- ☆34Oct 23, 2026Updated 2 weeks 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.
- Algebraic multigrid solvers in JAX☆43Aug 27, 2026Updated 2 months ago
- A complete example of batched refactorization in cuSOLVER.☆12Dec 22, 2021Updated 4 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆48Updated this week
- Calculation of Wigner symbols and related constants☆14Jul 27, 2026Updated 3 months ago
- Simple Xarray + JAX Integration☆21Mar 21, 2026Updated 7 months ago
- JAXMg: A multi-GPU linear solver in JAX☆113Updated this week
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆16May 20, 2024Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆143Updated this week
- A small library of paramaterizations and parameter constraints for PyTrees.☆68Jan 23, 2026Updated 9 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.
- Repository with notebooks about Physics Informed Neural Networks, written in JAX + Flax.☆11Nov 12, 2022Updated 3 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆587Oct 27, 2026Updated last week
- Matrix-free linear algebra in JAX.☆183Updated this week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last month
- Generate Streamlit apps from Tesseracts, automagically.☆17Updated this week
- Graphs defined by a grid of vertices. Compatible with the Graphs.jl ecosystem☆12Oct 28, 2026Updated last week
- ☆13May 14, 2025Updated last year
- ☆26Oct 10, 2026Updated 3 weeks ago
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆15Oct 30, 2026Updated last week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆23Oct 12, 2023Updated 3 years ago
- Mathematical Programming in JAX☆174Jun 2, 2025Updated last year
- A simple one file Python project to run 2D phase field problems with linear quadrilaterals☆24Sep 1, 2026Updated 2 months ago
- An interface for using NVIDIA's AMGX library from the Julia language.☆21Updated this week
- A simulator for ecological dynamics written in Julia.☆27Updated this week
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated last year
- Differentiable QP solver in JAX☆177Jul 24, 2026Updated 3 months ago
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Sep 17, 2026Updated last month
- Topology Optimization of Photonic Components using a Shape Library☆12Dec 20, 2024Updated last year
- 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☆46Oct 30, 2026Updated last week
- StateSpaceLearning.jl is a Julia package for time-series analysis using state space learning framework.☆21Sep 22, 2026Updated last month
- Algorithms for inverse design☆15Oct 21, 2025Updated last year
- Differentiable Pairing using Alignment-based Language Models☆27Jan 5, 2024Updated 2 years ago
- A lightweight podcast CMS☆10Apr 12, 2023Updated 3 years ago
- Differentiable Hydrodynamical Simulations☆18Oct 27, 2026Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆83Feb 12, 2026Updated 8 months ago