TorchFSM: Fourier Spectral Method with PyTorch
☆55Feb 23, 2026Updated 2 weeks ago
Alternatives and similar repositories for torchfsm
Users that are interested in torchfsm are comparing it to the libraries listed below
Sorting:
- Efficient Differentiable n-d PDE Solvers in JAX.☆155Mar 2, 2026Updated last week
- [NeurIPS 23] Official Code for "Learning Efficient Surrogate Dynamic Models with Graph Spline Networks"☆16Jul 19, 2024Updated last year
- ☆19Nov 24, 2024Updated last year
- Neural Emulator Architectures in JAX.☆23Feb 20, 2026Updated 2 weeks ago
- Official implementation of the paper "Light Transport-aware Diffusion Posterior Sampling for Single View Reconstruction of Volumes"☆17Aug 1, 2025Updated 7 months ago
- ☆47Feb 13, 2026Updated 3 weeks ago
- A simple Physics Informed Neural Network (PINN) implementation in TensorFlow 2 for magnetic field mapping using finite amount of magnetic…☆11Jun 7, 2022Updated 3 years ago
- Official code for paper: Conservative objective models are a special kind of contrastive divergence-based energy model☆14Aug 15, 2023Updated 2 years ago
- ☆11May 12, 2023Updated 2 years ago
- Chemical Data Science and Engineering - University of Toronto☆22Nov 21, 2025Updated 3 months ago
- Introductory workshop on using deep neural networks for symbolic regression☆16Feb 8, 2026Updated last month
- The code of the paper: M. Karami, “HiGen: Hierarchical Graph Generative Networks”, arXiv preprint arxiv:2305.19337☆10Apr 9, 2024Updated last year
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 years ago
- ☆31Sep 13, 2025Updated 5 months ago
- A variational auto-encoder (VAE) framework with a new type of prior "Variational Mixture of Posteriors" prior, or VampPrior for short.☆10Apr 7, 2021Updated 4 years ago
- Public Repo for the paper "Overcoming The Spectral-Bias of Neural Value Approximation"☆11May 25, 2024Updated last year
- ☆54Jul 21, 2025Updated 7 months 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 data-driven framework for the identification of latent port-Hamiltonian systems☆19Updated this week
- Prediction of Fluid Flow in Porous Media by Sparse Observations and Physics-Informed PointNet☆14Aug 24, 2024Updated last year
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 8 months ago
- ModelFLOWs application☆22Feb 11, 2025Updated last year
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆109Dec 13, 2024Updated last year
- This repository contains code for the paper "MAgNet: Mesh-Agnostic Neural PDE Solver" https://arxiv.org/abs/2210.05495☆38Jun 21, 2023Updated 2 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- Differentiable neural-network solver for data assimilation of ice shelves written in JAX☆39May 12, 2025Updated 9 months ago
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆18Jan 6, 2026Updated 2 months ago
- ☆18Aug 9, 2022Updated 3 years ago
- PhysiOpt: Physics-Driven Shape Optimization for 3D Generative Models☆36Nov 18, 2025Updated 3 months ago
- Physics-informed neural networks☆16Nov 26, 2020Updated 5 years ago
- ☆12Aug 22, 2025Updated 6 months ago
- An interpretable data-driven framework for building generative reduced order models with embedded uncertainty quantification☆48Feb 17, 2026Updated 2 weeks ago
- ☆26Jun 12, 2025Updated 8 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- ☆20Dec 2, 2024Updated last year
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- ☆28Jan 19, 2026Updated last month
- ☆36Jul 11, 2017Updated 8 years ago
- Official implementation of Scalable Transformer for PDE surrogate modelling☆54Mar 23, 2024Updated last year