PhiFlow
github.com/tum-pbs/phiflowDifferentiable PDE solving framework for machine learning with built-in fluid simulation, supporting PyTorch/JAX/TensorFlow backends and enabling neural network training within physical simulations (TUM, MIT License)
Sourced from
- Awesome AI for Science — github.com/tum-pbs/phiflow
- GitHub — github.com/tum-pbs/phiflow
Related resources
Efficient differentiable n-dimensional PDE solvers built on JAX and Equinox, shipping 46+ built-in equations with Fourier spectral methods, exponential time differencing, and full auto-differentiation for physics-based deep learning workflows (MIT, 200+ stars, 2024)
Fast, differentiable, JIT-free finite element library for PyTorch enabling GPU-native PDE solving with native autograd, tensorized assembly, and sparse linear algebra; part of the TensorGalerkin framework (218+ stars, Apache 2.0)
Machine learning in Julia
Numerical differential equation solving in JAX
Machine learning toolkit for many-body quantum systems, implementing neural quantum states, variational Monte Carlo, and tensor network algorithms to solve ground-state and dynamical problems in condensed matter physics and quantum chemistry (EPFL & collaborators, Nature Physics 2019/2022+, 670+ stars)
Multi-type data labeling and annotation tool