Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms, and mixed spectral bases. Some notebooks are presented: https://spectraldns.github.io/jaxfun/
☆20Jun 24, 2026Updated 3 weeks ago
Alternatives and similar repositories for jaxfun
Users that are interested in jaxfun are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated last month
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- GPU-accelerated differentiable sparse linear solver library for JAX☆33Updated this week
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 8 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆13Jun 1, 2026Updated last month
- Efficient GPU-computing simulation platform for differentiable phase field model☆25Apr 14, 2026Updated 3 months ago
- A GPU-accelerated and differentiable library for landscape genetics and ecological connectivity modelling.☆13Jun 4, 2026Updated last month
- Website for the book "The Elements of Differentiable Programming".☆15Jul 2, 2025Updated last year
- ☆14Feb 18, 2024Updated 2 years ago
- A fast, differentiable, JIT-free, debugging-friendly finite element library for PyTorch.☆185Jul 10, 2026Updated last week
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- About code release of "Learning, Solving and Optimizing PDEs with TensorGalerkin: an efficient high-performance Galerkin assembly algorit…☆20Feb 10, 2026Updated 5 months ago
- Neural Emulator Architectures in JAX.☆25Jun 1, 2026Updated last month
- 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.
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆16May 20, 2024Updated 2 years ago
- my personal guidelines writing papers and managing the publication cycle - things that come up again and again☆21Apr 15, 2026Updated 3 months ago
- High performance computational platform in Python for the spectral Galerkin method☆232Jun 22, 2026Updated 3 weeks ago
- Differentiable Hydrodynamical Simulations☆17Updated this week
- Pseudospectral Kolmogorov Flow Solver☆43Oct 24, 2023Updated 2 years ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆45Nov 21, 2025Updated 7 months ago
- FEniCSx finite element basis evaluation library☆142Updated this week
- A package for estimating code complexity in Julia☆21May 6, 2026Updated 2 months ago
- Connect to Slurm Compute Nodes Directly from VSCode. Do your debugging and development easily.☆31Oct 8, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Universal components for differentiable scientific computing.☆115Updated this week
- Mdformat plugin to Blacken Python code blocks☆28Mar 9, 2025Updated last year
- Orthogonal polynomials with JAX☆29Jun 22, 2026Updated 3 weeks ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆26Jun 2, 2026Updated last month
- SParse lINear solvers in JAX☆48Updated this week
- pyPFC: An Open-Source Python Package for Phase Field Crystal Simulations☆16Jun 25, 2026Updated 3 weeks ago
- Code for 'DiffuMeta: Algebraic Language Models for Inverse Design of Metamaterials via Diffusion Transformers'☆34Apr 23, 2026Updated 2 months ago
- ☆58Feb 19, 2025Updated last year
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆34May 30, 2025Updated last year
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆70Jul 1, 2026Updated 2 weeks ago
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- ☆15Oct 26, 2021Updated 4 years ago
- Tools for representation, morphologic characterization, and hemodynamics analysis of flow in saccular aneurysms☆12Mar 22, 2026Updated 3 months ago
- JAXMg: A multi-GPU linear solver in JAX☆58Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆136Jul 6, 2026Updated last week
- Create "fly-throughs" in 3d visualization☆16Updated this week