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)
Machine learning in Julia
Numerical differential equation solving in JAX
Physics-Informed Neural networks for Advanced modeling in PyTorch
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