Aeroscience-Computations-Analysis-Lab/underPINN
A modular, JAX-based framework for Physics-Informed Neural Networks, designed for scalable PDE-constrained learning with support for domain decomposition, attention mechanisms, and performance-oriented training.
GitHub repository with 22 stars and 6 forks.
Language: Python
Topics: deep-learning, jax, neural-network, physics-informed-neural-networks, pinn, pytorch, scientific-machine-learning, tensorflow