Neural Emulator Architectures in JAX.
☆23Feb 20, 2026Updated 2 weeks ago
Alternatives and similar repositories for pdequinox
Users that are interested in pdequinox are comparing it to the libraries listed below
Sorting:
- Efficient Differentiable n-d PDE Solvers in JAX.☆155Updated this week
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆99Nov 11, 2025Updated 3 months ago
- A minimal JAX library for connectivity modelling at scale☆11Jan 27, 2026Updated last month
- Simple differentiable approximate ocean models built with JAX.☆16Oct 13, 2023Updated 2 years ago
- ☆14Feb 18, 2024Updated 2 years ago
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 8 months ago
- Special functions in Jax☆17Mar 1, 2025Updated last year
- Chrome browser extension that adds customizable keyboard shortcuts to the Overleaf online latex editor.☆13Jun 22, 2024Updated last year
- Transformer creation from scratch using Jax.☆19Dec 8, 2024Updated last year
- Pseudospectral Kolmogorov Flow Solver☆42Oct 24, 2023Updated 2 years ago
- Finite difference tools in JAX☆23Feb 4, 2024Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19May 25, 2024Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated 2 weeks ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- TorchFSM: Fourier Spectral Method with PyTorch☆55Feb 23, 2026Updated last week
- ☆36Jul 11, 2017Updated 8 years ago
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Nov 3, 2025Updated 4 months ago
- GCM Physics written in JAX☆85Updated this week
- NeuralGCM's differentiable dycore☆317Feb 27, 2026Updated last week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆34Apr 19, 2025Updated 10 months ago
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆76Nov 21, 2025Updated 3 months ago
- Jaxpr Visualisation Tool☆36Dec 22, 2024Updated last year
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆15Aug 30, 2025Updated 6 months ago
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆43Apr 16, 2025Updated 10 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- Flood Mapping Intercomparison☆16Nov 5, 2025Updated 4 months ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆41Nov 21, 2025Updated 3 months ago
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆95Feb 27, 2026Updated last week
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Feb 10, 2026Updated 3 weeks ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆70Jan 1, 2025Updated last year
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- Use Fourier transform to learn operators in differential equations.☆45Dec 1, 2020Updated 5 years ago
- A hands-on tutorial on supervised learning with Gaussian processes☆37Nov 22, 2019Updated 6 years ago
- Codes for Numerical Analysis course☆39Apr 22, 2025Updated 10 months ago
- Official implementation of the paper "Light Transport-aware Diffusion Posterior Sampling for Single View Reconstruction of Volumes"☆17Aug 1, 2025Updated 7 months ago
- MIT PhD Thesis template☆11May 22, 2025Updated 9 months ago