srigas / jaxKAN
A JAX-based implementation of Kolmogorov-Arnold Networks
☆47Updated last month
Related projects ⓘ
Alternatives and complementary repositories for jaxKAN
- Code for the paper Universal Physics Transformers☆78Updated last month
- An unofficial implementation of the Fourier Neural Operator in Flax☆16Updated 5 months ago
- Spectral Neural Operator☆73Updated 11 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆75Updated 2 weeks ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆104Updated last month
- Convolutional Differential Operators for Physics-based Deep Learning Study☆24Updated 3 months ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆33Updated 9 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆34Updated 3 weeks ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆126Updated 6 months ago
- ☆233Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- ☆110Updated 3 weeks ago
- Code of ICML paper arxiv.org/abs/2302.08105☆14Updated last year
- Datasets and code for results presented in the ProbConserv paper☆51Updated 5 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆48Updated last year
- Example problems in Physics informed neural network in JAX☆76Updated last year
- ☆231Updated last week
- Python Algorithms for Randomized Linear Algebra☆46Updated last year
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆19Updated last week
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Interpolation and function approximation with JAX☆136Updated 3 weeks ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆70Updated 8 months ago
- Deep Bayesian Optimization for Problems with High-Dimensional Structure☆14Updated 2 years ago
- Better implementation of Kolmogorov Arnold Network☆21Updated 5 months ago
- Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML☆37Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- Code for the book "The Elements of Differentiable Programming".☆65Updated 3 months ago
- Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆20Updated last month
- Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"☆43Updated 5 months ago
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆26Updated 2 months ago