NVIDIA / physicsnemo

git clone https://github.com/NVIDIA/physicsnemo.git

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods