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☆53Aug 4, 2026Updated last month
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated last month
- Automatic Sparse Differentiation in JAX.☆47Updated this week
- GPU-accelerated differentiable sparse linear solver library for JAX☆50Aug 30, 2026Updated last week
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆70Aug 1, 2026Updated last month
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Schedule free optimiser implemented in JAX using Optimistix☆14May 29, 2024Updated 2 years ago
- 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☆47Updated this week
- A small library of paramaterizations and parameter constraints for PyTrees.☆66Jan 23, 2026Updated 7 months ago
- ☆18Jul 25, 2026Updated last month
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆14Jun 4, 2026Updated 3 months ago
- Matrix-free linear algebra in JAX.☆180Updated this week
- Generate Streamlit apps from Tesseracts, automagically.☆17Updated this week
- 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.
- 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
- ☆26Apr 27, 2026Updated 4 months ago
- Simple Xarray + JAX Integration☆21Mar 21, 2026Updated 5 months ago
- A simulator for ecological dynamics written in Julia.☆27Aug 15, 2026Updated 3 weeks ago
- Differentiable Pairing using Alignment-based Language Models☆27Jan 5, 2024Updated 2 years ago
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 10 months ago
- Parallel solution of structured nonlinear optimization problems☆13Dec 16, 2025Updated 8 months ago
- An unofficial Julia wrapper for the RAPIDS.ai ecosystem using PythonCall.jl☆23Apr 21, 2024Updated 2 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆20Updated this week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆82Feb 12, 2026Updated 6 months ago
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Updated this week
- Particle-mesh based calculations of long-range interactions in JAX☆30Aug 9, 2026Updated 3 weeks ago
- Universal components for differentiable scientific computing.☆132Updated this week
- Automated material constitutive modeling in JAX☆45Jul 23, 2026Updated last month
- Differentiable Hydrodynamical Simulations☆17Jul 17, 2026Updated last month
- Mathematical Programming in JAX☆170Jun 2, 2025Updated last year
- ☆36Jul 19, 2023Updated 3 years ago
- 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.
- Coloring algorithms for sparse Jacobian and Hessian matrices☆45Updated this week
- JAX translation of boltz☆31Aug 11, 2026Updated 3 weeks ago
- Calculation framework☆13Jan 20, 2026Updated 7 months ago
- ☆41Jun 29, 2026Updated 2 months ago
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆58Updated this week
- Cut finite elements for FEniCSx☆17Jun 24, 2026Updated 2 months ago
- Automatically exported from code.google.com/p/another-chebpy☆13Apr 28, 2015Updated 11 years ago