nvidia/physicsnemo

NVIDIA PhysicsNeMo

PyTorch로 물리 AI 모델을 만들고, 훈련시키고, 확장하세요.

문서 · 예시 · 모델 · 블로그 · 토론 · 기여

NVIDIA PhysicsNeMo는 물리 머신러닝을 위한 오픈소스 PyTorch 프레임워크입니다 (물리학 머신러닝), 과학적 머신러닝(SciML), 그리고 과학 및 공학을 위한 AI 분야입니다. 재사용 가능한 라이브러리 구성 요소와 종단 간 교육 레시피를 제공합니다.

CUDA 12의 경우, 기본 설치, 선택 기능 또는 소스 설정은 다음을 참조하세요. 설치 옵션.

물리학 니모 액션

공기역학 기상 예보

통합 외부 공기역학: 현재 표면 및 부피 모델을 훈련 및 비교 StormCast : 생성형 지역 기상 예보

반도체 패키징 구조역학

언더필 디스펜싱: GeoTransolver와의 일시 에폭시 인터페이스 예측 충돌 역학: 변형 메쉬 위의 일시적 대리

지구물리학 의료

확산 FWI : 유도 생성 전파형 반전 심혈관 흐름: MeshGraphNet을 이용한 저차 예측

데이터 센터 열 설계 적층 제조

NVIDIA PhysicsNeMo

Build, train, and scale physics AI models with PyTorch.

Documentation · Examples · Models · Blog · Discussions · Contributing

NVIDIA PhysicsNeMo is an open-source PyTorch framework for physics machine learning (physics ML), scientific machine learning (SciML), and AI for science and engineering. It provides reusable library components and end-to-end training recipes.

pip install "nvidia-physicsnemo[cu13]"

For CUDA 12, a basic install, optional features, or source setup, refer to installation options.

PhysicsNeMo in Action

Aerodynamics
Weather Forecasting