NVIDIA PhysicsNeMo

NVIDIA PhysicsNeMo is an open-source deep-learning framework for building, training, fine-tuning and inferring Physics AI models using state-of-the-art SciML methods for AI4science and engineering.

PhysicsNeMo provides python modules to compose scalable and optimized training and inference pipelines to explore, develop, validate and deploy AI models that combine physics knowledge with data, enabling real-time predictions.