PyPI page
Home page
Author:
NVIDIA PhysicsNeMo Team
Summary:
A deep learning framework for AI-driven multi-physics systems
Latest version:
2.0.0
Required dependencies:
cftime
|
einops
|
gitpython
|
h5py
|
hydra-core
|
importlib-metadata
|
jaxtyping
|
numpy
|
nvtx
|
omegaconf
|
onnx
|
packaging
|
pandas
|
requests
|
s3fs
|
tensordict
|
termcolor
|
timm
|
torch
|
torchvision
|
tqdm
|
treelib
|
warp-lang
Optional dependencies:
cuml-cu12
|
cuml-cu13
|
cupy-cuda12x
|
cupy-cuda13x
|
dask
|
line-profiler
|
matplotlib
|
mlflow
|
natten
|
netcdf4
|
nvidia-dali-cuda120
|
nvidia-dali-cuda130
|
pyacvd
|
pylibraft-cu12
|
pylibraft-cu13
|
pyvista
|
scipy
|
stl
|
tensordict
|
tfrecord
|
torch
|
torch-cluster
|
torch-geometric
|
torch-scatter
|
torch-sparse
|
torchvision
|
transformer-engine
|
vtk
|
wandb
|
xarray
|
zarr
Downloads last day:
599
Downloads last week:
7,527
Downloads last month:
39,261