srigas / jaxKAN
A JAX-based implementation of Kolmogorov-Arnold Networks
☆44Updated last month
Related projects ⓘ
Alternatives and complementary repositories for jaxKAN
- An unofficial implementation of the Fourier Neural Operator in Flax☆16Updated 5 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆73Updated this week
- Spectral Neural Operator☆71Updated 10 months ago
- Code for the paper Universal Physics Transformers☆74Updated last month
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆100Updated last month
- Datasets and code for results presented in the ProbConserv paper☆51Updated 4 months ago
- Multiple dispatch over abstract array types in JAX.☆102Updated 3 weeks ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆122Updated 6 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆47Updated last year
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆16Updated this week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆33Updated last week
- Matrix-free linear algebra in JAX.☆101Updated last month
- Convolutional Differential Operators for Physics-based Deep Learning Study☆23Updated 3 months ago
- Code for the book "The Elements of Differentiable Programming".☆63Updated 2 months ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆32Updated 9 months ago
- This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs☆128Updated 3 weeks ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆70Updated 7 months ago
- Code of ICML paper arxiv.org/abs/2302.08105☆14Updated last year
- ☆19Updated 2 years ago
- ☆231Updated last month
- Implementation of Forward Laplacian algorithm in JAX☆55Updated this week
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆22Updated this week
- ☆29Updated last year
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆57Updated 4 months ago
- Stencil computations in JAX☆66Updated last year
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated 5 months ago
- [ICLR 2024] Scaling physics-informed hard constraints with mixture-of-experts.☆23Updated 4 months ago
- ☆37Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆120Updated last month
- Practicum on Supervised Learning in Function Spaces☆32Updated 2 years ago