nvidia-modulus

View on PyPIReverse Dependencies (4)

0.9.0 nvidia_modulus-0.9.0-py3-none-any.whl

Wheel Details

Project: nvidia-modulus
Version: 0.9.0
Filename: nvidia_modulus-0.9.0-py3-none-any.whl
Download: [link]
Size: 548782
MD5: 1ac7ff2420b94a2e0d2ea3a8e4cd7f07
SHA256: e43956718629c02adc5dd30646d11845fe04086225a4df98ac4a106ed78dec78
Uploaded: 2024-11-27 17:28:25 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: nvidia-modulus
Version: 0.9.0
Summary: A deep learning framework for AI-driven multi-physics systems
Author: NVIDIA Modulus Team
Project-Url: Homepage, https://github.com/NVIDIA/modulus
Project-Url: Documentation, https://docs.nvidia.com/modulus/#core
Project-Url: Issues, https://github.com/NVIDIA/modulus/issues
Project-Url: Changelog, https://github.com/NVIDIA/modulus/blob/main/CHANGELOG.md
License: Apache 2.0
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.10
Requires-Dist: torch (>=2.0.0)
Requires-Dist: numpy (<1.25,>=1.22.4)
Requires-Dist: xarray (>=2023.1.0)
Requires-Dist: zarr (>=2.14.2)
Requires-Dist: fsspec (>=2023.1.0)
Requires-Dist: s3fs (>=2023.5.0)
Requires-Dist: nvidia-dali-cuda120 (>=1.35.0)
Requires-Dist: setuptools (>=67.6.0)
Requires-Dist: certifi (>=2023.7.22)
Requires-Dist: pytz (>=2023.3)
Requires-Dist: treelib (>=1.2.5)
Requires-Dist: tqdm (>=4.60.0)
Requires-Dist: nvtx (>=0.2.8)
Requires-Dist: onnx (>=1.14.0)
Requires-Dist: timm (>=0.9.12)
Requires-Dist: h5py (>=3.7.0); extra == "all"
Requires-Dist: netcdf4 (>=1.6.3); extra == "all"
Requires-Dist: ruamel.yaml (>=0.17.22); extra == "all"
Requires-Dist: scikit-learn (>=1.0.2); extra == "all"
Requires-Dist: scikit-image (>=0.24.0); extra == "all"
Requires-Dist: warp-lang (>=1.0); extra == "all"
Requires-Dist: vtk (>=9.2.6); extra == "all"
Requires-Dist: pyvista (>=0.40.1); extra == "all"
Requires-Dist: cftime (>=1.6.2); extra == "all"
Requires-Dist: einops (>=0.7.0); extra == "all"
Requires-Dist: pyspng (>=0.1.0); extra == "all"
Requires-Dist: shapely (>=2.0.6); extra == "all"
Requires-Dist: nvidia-modulus[launch]; extra == "all"
Requires-Dist: nvidia-modulus[dev]; extra == "all"
Requires-Dist: nvidia-modulus[makani]; extra == "all"
Requires-Dist: nvidia-modulus[fignet]; extra == "all"
Requires-Dist: pytest (>=6.0.0); extra == "dev"
Requires-Dist: pyyaml (>=6.0); extra == "dev"
Requires-Dist: black (==22.10.0); extra == "dev"
Requires-Dist: interrogate (==1.5.0); extra == "dev"
Requires-Dist: coverage (==6.5.0); extra == "dev"
Requires-Dist: ruff (==0.0.290); extra == "dev"
Requires-Dist: jaxtyping (>=0.2); extra == "fignet"
Requires-Dist: torch-scatter (>=2.1); extra == "fignet"
Requires-Dist: torchinfo (>=1.8); extra == "fignet"
Requires-Dist: warp-lang (>=1.0); extra == "fignet"
Requires-Dist: webdataset (>=0.2); extra == "fignet"
Requires-Dist: hydra-core (>=1.2.0); extra == "launch"
Requires-Dist: termcolor (>=2.1.1); extra == "launch"
Requires-Dist: wandb (>=0.13.7); extra == "launch"
Requires-Dist: mlflow (>=2.1.1); extra == "launch"
Requires-Dist: pydantic (>=2.4.2); extra == "launch"
Requires-Dist: imageio (>=2.28.1); extra == "launch"
Requires-Dist: moviepy (>=1.0.3); extra == "launch"
Requires-Dist: torch-harmonics (<0.7.1,>=0.6.5); extra == "makani"
Requires-Dist: tensorly (>=0.8.1); extra == "makani"
Requires-Dist: tensorly-torch (>=0.4.0); extra == "makani"
Provides-Extra: all
Provides-Extra: dev
Provides-Extra: fignet
Provides-Extra: launch
Provides-Extra: makani
Description-Content-Type: text/markdown
License-File: LICENSE.txt
[Description omitted; length: 18785 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.5.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
modulus/__init__.py sha256=lTuRRH7pfJ73O1eb3lvlpDEkxD4HRefjA5ad98PcJZ8 899
modulus/constants.py sha256=p8Z2v7St9FlDXMf-AOnYJmWMkoRonqh5An_Y3UOuOwk 1401
modulus/datapipes/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/datapipes/datapipe.py sha256=eXEtpytLqexfKhA3iaiQ996H7qHd_oOUeKlpNuJabDA 2124
modulus/datapipes/meta.py sha256=-8SYSIWTDmLEVxmIJClLcNlGk0KyLaeRDAVHP4vKe4o 1055
modulus/datapipes/benchmarks/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/datapipes/benchmarks/darcy.py sha256=tbn-5_JtqBuddOWaG7txV1EL0xTDHl37DO3np7y4YS8 11787
modulus/datapipes/benchmarks/kelvin_helmholtz.py sha256=8wJo3Gweq6lXlhCXZUW44Gh2HGiMasCpkAI5bL8ftC8 16764
modulus/datapipes/benchmarks/kernels/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/datapipes/benchmarks/kernels/finite_difference.py sha256=LOaUW5Pl6IMuniJ4Jzq9yEVgRRuSMbFBY065a89QgHM 4357
modulus/datapipes/benchmarks/kernels/finite_volume.py sha256=pZGXwmzu1PzGruWi-sODEIN3lAIre4oLBZlZ8vKIKHI 20604
modulus/datapipes/benchmarks/kernels/indexing.py sha256=zrBN1WG-zkPtdu19Y8eo2h1TWxoZfpj4v-9-ahe1rtA 3752
modulus/datapipes/benchmarks/kernels/initialization.py sha256=0GLXg0LbGHttyz_rLU6jh98jRkANPDZqY81waxDc1eM 2148
modulus/datapipes/benchmarks/kernels/utils.py sha256=fiwUYImiDg8eQFa6XUrhBD25T5uR3HqTdx5OqZ41y2Y 4340
modulus/datapipes/cae/__init__.py sha256=xS-qbToIB_UHk10_vELtwjyJSaGqFx8Mv6wi3cy_HzA 757
modulus/datapipes/cae/mesh_datapipe.py sha256=m1Jf01OzzABgM35gPGitJmRnKsWcineR3fvFLue2lwk 16596
modulus/datapipes/cae/readers.py sha256=Tv5XfpYJatvJYF4GLm_JJA9VKJN4Ddg408CHoRvz94Q 5144
modulus/datapipes/climate/__init__.py sha256=09waL7VcflHVa0ntKQFpU6Ycw0Ej2J8oR3KeXZWPGzg 892
modulus/datapipes/climate/climate.py sha256=P2KbSRumJi7cThyQV9nOMHd7AiDHZAKdzShOy2oEKkg 32880
modulus/datapipes/climate/era5_hdf5.py sha256=GKzdHR92pj-C6yg5bCE1C-2vH8YyjFQb_Rbrr2V2kFo 23839
modulus/datapipes/climate/era5_netcdf.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/datapipes/climate/synthetic.py sha256=ZMWE9edEUz_Ls8vYcNJ8VhjHcOYovrvDEq9rE7zMMQc 6933
modulus/datapipes/climate/utils/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/datapipes/climate/utils/invariant.py sha256=wVIZhrRVW1xINISd9tfvoJSjbUc-CgUDTQkzm4Ub0MY 4656
modulus/datapipes/climate/utils/zenith_angle.py sha256=38QHshcgHVka9muealefvfNYT7u1gONoDwwHDhv1eRY 6955
modulus/datapipes/gnn/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/datapipes/gnn/ahmed_body_dataset.py sha256=GU03bgnMOrvLsKRw7J-FyAdaq8Nm_K9FVcgieNFfe7k 21937
modulus/datapipes/gnn/bsms.py sha256=Zmj9IgAM93qfIIh1roeicw1BByILV3UGsaLepTuuR6s 26571
modulus/datapipes/gnn/drivaernet_dataset.py sha256=1rUi9QkmgN5a7PojPFDSAZ6Q-zMO0AJuvKOfT1DWQ-M 14861
modulus/datapipes/gnn/lagrangian_dataset.py sha256=jcVdvm_jzhpdMLlpxdmmyoCuezBzHcK0gaNboEK1PzA 13022
modulus/datapipes/gnn/lagrangian_reading_utils.py sha256=SqKiSH1_EFU_mHmmeN1VjDpNdd4GXRLgOSmQRaW-rng 5384
modulus/datapipes/gnn/stokes_dataset.py sha256=UKQ8wUa4TSFh0k0dpcL3wjw4fMht9l_APYb00EIv_pI 11210
modulus/datapipes/gnn/utils.py sha256=mp4dRCBu1ZJHfpma5nBa1uYb3_778mkGoVW8PojJw4I 2660
modulus/datapipes/gnn/vortex_shedding_dataset.py sha256=MlVpXKm9SsJ2lTA8ReAk4UQOQY9q-1tdgjwGljsTf8I 15548
modulus/datapipes/gnn/vortex_shedding_re300_1000_dataset.py sha256=b3OI5sUHvu-c79se7vhd63FgzXqo42UkXoAeD5urBZk 8806
modulus/datapipes/healpix/__init__.py sha256=QvDf6l3HWAwiPMAZ9P627YS7j45wJeQRU5cZ_Kz4A40 771
modulus/datapipes/healpix/coupledtimeseries_dataset.py sha256=L-3E6VwxnDMaSymAB2e-zoXYfk0zmpk9iph_ZNPI5NI 14208
modulus/datapipes/healpix/couplers.py sha256=CeSMCzVM14kBVyvXN7SHBewGBqZ0mcxcfuoe5cRrR74 19777
modulus/datapipes/healpix/data_modules.py sha256=gpscirQdbDYJg20HqO0o3Gw2Kx5jUZXPkvQN3Au5YFc 42311
modulus/datapipes/healpix/timeseries_dataset.py sha256=FuXHCvKxNgnf1ICDUxiJJ3RTZAH0_lr2Ex1Rqt_LfOQ 16294
modulus/deploy/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/deploy/onnx/__init__.py sha256=K8k4JhR6o021VY4ugoqmlrD-xhReL1b4odC9SV1goK4 778
modulus/deploy/onnx/utils.py sha256=CfoQVWwU4dVGIaeTQSujv1eYW8qo43gZFIDVehHzgXU 4769
modulus/deploy/triton/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/deploy/trt/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/distributed/__init__.py sha256=OebmpsHCMS3HoE6nqmqa-c7ABrKjS81uMkhX108dFw0 1081
modulus/distributed/autograd.py sha256=wy1sUKwwIQB4z1PlqhljA8v8A0kvsam2zy7w_qaSRLQ 15152
modulus/distributed/config.py sha256=IvFDzheedFu9vvz-w8rpb65p0-YsKtr-vPY1PPwwhAg 8163
modulus/distributed/fft.py sha256=A84kAh3nHKwtbe6v-iihJBudrWeELB1yEaaQiyVDUyQ 7171
modulus/distributed/manager.py sha256=LxT1-JUxt3iUQZym7-UDs5aJ8ywepushEaHaRQZELtY 22142
modulus/distributed/mappings.py sha256=3SHrFROLjp8S4COJwVWDonhg1ZoTCeagtiXYKwuS92Q 4583
modulus/distributed/utils.py sha256=nKxDUJCIBrCDwccKo4ynOHBJXfMOTv3AYy9I0gbdygY 25926
modulus/experimental/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/launch/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/launch/config/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/launch/logging/__init__.py sha256=wPaU4YekMfom3OtsxdrgMBEbQWXqr2Ht9pINV4UotgY 882
modulus/launch/logging/console.py sha256=XmOEt3sy9NDy8jdQhGPJtfOU9lWfmYFNVCuNXofxpZ0 2734
modulus/launch/logging/launch.py sha256=i5LEZTTJPGaHJOPkMVtaUYwf0ggRhkVN-CWvkARhTGs 14870
modulus/launch/logging/mlflow.py sha256=kQJAu3CsAN-smOJxGfnM07MfWE-7B7mcAcZc616rFSg 7173
modulus/launch/logging/utils.py sha256=Iig4lQL51VsYzWrXb_ZPyxEm0ELxcCPIWJ-NyfgNMQA 2036
modulus/launch/logging/wandb.py sha256=aI1ePVEdEFESHIB5tcZUsB5guhWPJQUI-QTsusv2oF8 4486
modulus/launch/utils/__init__.py sha256=PA_rLI0eJ79Gmm4WZnjouOA_r7eunztMYiaFDb4bdSU 774
modulus/launch/utils/checkpoint.py sha256=qjG109r329Oz3sdcyoKVuPRA9xxXiCEcTBYeqIq6Eec 14386
modulus/metrics/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/metrics/cae/cfd.py sha256=8vydYCrMQdEqkJDMS4gmffB-r9kHELG6pF1CPr8Eers 8684
modulus/metrics/cae/integral.py sha256=Kwj_fJLUrdea--6Kqc5RqdJzNU89k_SNjnDdFuz6HVA 3805
modulus/metrics/climate/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/metrics/climate/acc.py sha256=bTHUeo4DUmMehHsUQLNNmsPT4J_jxsJP08_zN2COkNQ 2830
modulus/metrics/climate/efi.py sha256=DpCgfC3qhpw7i824AL6poYetvZWfbS4NLGBdbGMEyzI 4601
modulus/metrics/climate/healpix_loss.py sha256=_emiojXmrWBuTEkqbHsKC03W3T-DqRLA0QX-Wye8eG4 6329
modulus/metrics/climate/loss.py sha256=NFSHfgnfjBfXM4W_wRdWjXClJJ-jXmU6dYxnn6DrQCk 13543
modulus/metrics/climate/reduction.py sha256=hmQaYORhxR454GshbooCIv5uX_Zrm1Jdpg1V7Fok1II 5599
modulus/metrics/diffusion/__init__.py sha256=ZJ1tDXUZiaXuU9Z5oC0BbuD2UspGrk2Gqrl09W6rbTI 915
modulus/metrics/diffusion/fid.py sha256=jl3OT2U4eJDBfuVE_KNK1jW9tgQqszBp6Nozmg2tya4 1886
modulus/metrics/diffusion/loss.py sha256=wXictBh9QuK7XJ-KxtNoiGDCYzLMUMOLm3gGzTJW_ZA 30569
modulus/metrics/general/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/metrics/general/calibration.py sha256=px7RUDwX6y_0ARD6aEv1dKJX_7qnKdVQz4EfrphBTME 4986
modulus/metrics/general/crps.py sha256=1riDQg58AHq6TroM-iIChAXnXj9Dk8AA5ijmClB14ds 12273
modulus/metrics/general/ensemble_metrics.py sha256=wYJR47UK_piHnfPijvmJlKp_SvYXsUF8zK-SfwiZya0 13531
modulus/metrics/general/entropy.py sha256=wVFVIN74meTJwxgjsYuFFUHdSdI7UvdeOqjCprw0fDA 4813
modulus/metrics/general/histogram.py sha256=ZQwrgk5lNCqGaTSgvZ1bbxQMo9KhwimbR60OTJRTsLE 27525
modulus/metrics/general/mse.py sha256=s_OcQ4BAt-4ABlP0GNTrm_KrZYpLZGgkMIPhxitt9mM 1951
modulus/metrics/general/reduction.py sha256=jXUi0K86A_6uMFg9OyiqMPZqtz_-9XYtz2LkIUjGs0o 4401
modulus/metrics/general/wasserstein.py sha256=v-ddmoosTx7q70xlCU1JorhaCWdcno3EW8Qs4UY31Bc 5719
modulus/models/__init__.py sha256=S-ABX3XgPmfzkT7vB1nw9sVRbpPrgdx2hbDsROJpia4 744
modulus/models/fcn_mip_plugin.py sha256=unT5pi-FRAZorckl0crqxtOotEf6RgMtk5n5wJiC3YQ 11979
modulus/models/meta.py sha256=rCpHRa6Wq_CJpRlVVf9Nj6Vj7BgmIghvkQsjv8KwWXQ 1707
modulus/models/module.py sha256=C8NWojmem8jqfCQVpsSynQgLfc4LpzDaXnMQvHx08aY 15770
modulus/models/afno/__init__.py sha256=kswuITFCYsDquL_yeGkYW7pygfTN0JYKmXkuFanVxUo 810
modulus/models/afno/afno.py sha256=XQ1dIoM3ZLu8ymi9A1NIOU7OIvkCtGUXevMDCGWBdxo 18237
modulus/models/afno/modafno.py sha256=Y27iXsS8M8H31hOox5xzxa7DrVXwmHQqbNsnLdVWgXk 22735
modulus/models/afno/modembed.py sha256=LANtmyFxnP-erx3vAuEPRLfLUC5oMfefzkkpvR7SlRM 3663
modulus/models/afno/distributed/__init__.py sha256=1NIxpJHKEILoc3ZG94GrNn-iLlmaagnbiGSMsQwK1v8 751
modulus/models/afno/distributed/afno.py sha256=EyWq5pqtXoFTbpVHPQMPqkfAfbdX_DSrjilwaPpoR_w 11761
modulus/models/afno/distributed/layers.py sha256=JX3bxYl4bN3lc6VXQUrRih1zaAIhZNHfBolJWk6MBw4 15128
modulus/models/diffusion/__init__.py sha256=YE8CKqWGqHMkdl3U3fB9E_wKyPu-o6N6cCuODCVoqT4 1193
modulus/models/diffusion/dhariwal_unet.py sha256=jqo-t969J2CRiO6xXLKa1_oqBmGkqLaXu8ln-Aa4b2o 9293
modulus/models/diffusion/layers.py sha256=k3NT4uGA_2nmeHP2mwwJNQEnIhOrXnkYBBRVBmEmDrc 20429
modulus/models/diffusion/preconditioning.py sha256=uQXlm3UmxEUCgIi_gqjihtuSMbrGl3amELhRrfXYHnY 36465
modulus/models/diffusion/song_unet.py sha256=sMLGXkhMk3V1G99y-h05t4WGNyFK2HI0sAaP3_15Lrs 36062
modulus/models/diffusion/unet.py sha256=tMYRmP1WFzIMNpCFXOHFw4gb4AJYul1eHfDgpFsTqUY 5272
modulus/models/diffusion/utils.py sha256=OMIKF8aBeZ7iNjRq8GQVWoBilDwVtc5F6Ybj50sKGT8 2681
modulus/models/dlwp/__init__.py sha256=mSMtF8WhiDiw6I8LYGMYhxZt-qulz778w1ShmXv1jo8 740
modulus/models/dlwp/dlwp.py sha256=91oPPsOmbYvXz8wsLVQBuf2NUhXAUerzj67CbSKezHQ 13448
modulus/models/dlwp_healpix/HEALPixRecUNet.py sha256=73uf9rQsyzlh5zQ6IJKvj496C_tbPFKgfksbWtl2NRk 17710
modulus/models/dlwp_healpix/HEALPixUNet.py sha256=sKlid8YfZKzvTOPmXsjqN1rbMPseKuUxj5C7ly7bk2Q 14440
modulus/models/dlwp_healpix/__init__.py sha256=vaDX8A6wk2kV1EUf0rZmDm3C6Y--eqwWrcICKG6xoPQ 842
modulus/models/dlwp_healpix_layers/__init__.py sha256=cwDCZFzoYbwYh7Ahcor-rV91yW5cwTdtuVnuVP033xQ 1150
modulus/models/dlwp_healpix_layers/healpix_blocks.py sha256=73fK7V7R8tXo09pRSSe-YmmRw_leKNREgp0fgl6krzg 27735
modulus/models/dlwp_healpix_layers/healpix_decoder.py sha256=ASha3Jr0TJi77y_uVJ5PADd3j9zo36LPMPEGdri3sAQ 5800
modulus/models/dlwp_healpix_layers/healpix_encoder.py sha256=jwWJhuHCTCMD290FmMBSavRUpcEHn7GOvkcQ4IQ_VY8 4213
modulus/models/dlwp_healpix_layers/healpix_layers.py sha256=ijsXYKY2pj2rys_pjJ72lyczi3FUpLgS3yo385CaBjI 20656
modulus/models/fengwu/__init__.py sha256=hXOrg6pQXkpi0XUGEbq6_Yr2jgNFbBE45DUMJwb-5hQ 744
modulus/models/fengwu/fengwu.py sha256=zvpX14ANt6iVbB6BMVJK9gHkm0pUDK_2VjSkOAiGC1g 10445
modulus/models/figconvnet/base_model.py sha256=iegCePlJP25xLvmcuiuJK-YOc8AsFlQfTp0aXBUnVhk 1542
modulus/models/figconvnet/figconvunet.py sha256=3xqkcTqLejRROdWFljPhhg2draz5EtLBsC_Vnzu_iy0 14495
modulus/models/figconvnet/geometries.py sha256=8ItqXgE-4DmNxybtp__SzOLNYex0lLoJwzz81zLosH4 17658
modulus/models/figconvnet/grid_feature_group.py sha256=zNexTYcRcFNkmju-L-69tMHUAW-TD_I67HoSADeD5cc 20241
modulus/models/figconvnet/neighbor_ops.py sha256=sbgKlf5qQyY7aVBbfXXFD2DzRUze--fg3kdf6Zm0nnM 7099
modulus/models/figconvnet/point_feature_conv.py sha256=mtVMcwBLX62bSnifpBrxKuJBJ73BGP64ITjm34N784U 17713
modulus/models/figconvnet/point_feature_grid_conv.py sha256=AzHMopqwvoZLJvH01U-Uwc3stxVV0Lqyj6wrMJ5mL2U 11607
modulus/models/figconvnet/point_feature_grid_ops.py sha256=YqE9Q10rgvcR6npExOPF1LN_qvQk3QN6cta5haUuNMg 12035
modulus/models/figconvnet/warp_neighbor_search.py sha256=NXMmilcWUeqVzlOwKageeYSsvMcuVwxNPY80AoZf1pE 7867
modulus/models/figconvnet/components/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/models/figconvnet/components/encodings.py sha256=K6s37DWBLrrvPY_1MD9bQvc_w2bybHJ4hlJAhUizUGU 1645
modulus/models/figconvnet/components/mlp.py sha256=3o0zXQzlqah30Urm-HZ2mHlbTMAFYhBqb5CKrZ25RtA 5145
modulus/models/figconvnet/components/reductions.py sha256=6KlN7ILl9sPpAy7kkWKcyRSy42ZD7AvLM0E-VI6-A0M 2046
modulus/models/fno/__init__.py sha256=mOVRtGk0yA-FVvVliiyjlIn8gZMLzT0SIYx80UyVJx4 804
modulus/models/fno/fno.py sha256=egTvJ7dl9ak8nh4RxEUc-S5rlo9hkuyPMGlHuNTJVZ0 31980
modulus/models/gnn_layers/__init__.py sha256=q-AH1ThloRYuEJuh58XMrUP6B1Giq_ZXcrd4ylc51hs 931
modulus/models/gnn_layers/bsms.py sha256=dUGki1X9sUYocat7gG9DsXD68lDzJha7HytXxfgCgKw 23766
modulus/models/gnn_layers/distributed_graph.py sha256=XqpwndUFDtOEZW0LVo3rga5LYgK4deYJkfyxIyyHabk 40505
modulus/models/gnn_layers/embedder.py sha256=JM0n_dtRoSA38qYwxtpwoY8Hs14VlBjdGlVlC5thR6E 6252
modulus/models/gnn_layers/graph.py sha256=NtkwaE_pPM-ntbkTDzp5rwuWGKywusPB8rqRBp3sp3k 18402
modulus/models/gnn_layers/mesh_edge_block.py sha256=dLfmXPPTRPAgWkwpnoPfakDeVtfCqka7OJOlZyYwTek 3273
modulus/models/gnn_layers/mesh_graph_decoder.py sha256=5J8gFcvOvG_-MV1Kd1I4auCNeO_Z6gr4X8abvS_X8ME 4671
modulus/models/gnn_layers/mesh_graph_encoder.py sha256=iM9P6OGLwqL-2ASsIdxdECwtx3gzJRdHGytL2qfLaEg 5440
modulus/models/gnn_layers/mesh_graph_mlp.py sha256=A0eMqdK5WixK1jrtbWXbwlQ_Y7mffvtO2jkcE67xaJw 16004
modulus/models/gnn_layers/mesh_node_block.py sha256=aJNN-c76-3QgaGFhqay-uZ95imyA2SBx8ILBjX0ntaY 3330
modulus/models/gnn_layers/utils.py sha256=WHNlmIp5UBCmQaYgKD4FNT5ff0DILfrGlHGfckpROsw 13611
modulus/models/graphcast/__init__.py sha256=Rtg4_DQCoz_7tpTsLOhCmcX2Cc9eXf2xhzfK4oj784s 758
modulus/models/graphcast/graph_cast_net.py sha256=tjhZcOQUXj_YcXe5yBCnMhzSstOEYB6NSVa_0TkY-N8 36072
modulus/models/graphcast/graph_cast_processor.py sha256=qyeZRRMRmF68YBLhzkGdpy0_7AU0cQx_QOKJPB8uYTw 8207
modulus/models/layers/__init__.py sha256=AjtcNGWJBm336zznZp0UmxAwstbO1iYFFQiWRCdiV28 1636
modulus/models/layers/activations.py sha256=SYtM7oiL6PxI6QN_N69z6nrcgj9NopFnRIRJHF7U2KA 6294
modulus/models/layers/attention_layers.py sha256=sPP3EnlXLyDUwuyKvrgs4viaxgvehqYRzh9fL84nO2Y 8595
modulus/models/layers/conv_layers.py sha256=AH78dDpbVpelto6OZvZ_BxspnSVgExXM3R3oCmkmVXQ 3562
modulus/models/layers/dgm_layers.py sha256=XotF27djih4sgJBwtA0_GSw_BoQ5PsyINz1FVHu-BQs 3407
modulus/models/layers/drop.py sha256=Un3n5FPzuvGWTsFmmFV3qE5OBNKk12RZ0vTK2cB2Z1M 2363
modulus/models/layers/fft.py sha256=Fi9u6PvvbdjB-fCRpYrWod2HlixZxPblxSSkbGxqCeY 18138
modulus/models/layers/fourier_layers.py sha256=-N6VVK3C8bya9v4p-p7qDafI-_UlTl0LzQCA2rzTMj0 6079
modulus/models/layers/fully_connected_layers.py sha256=QDEmK-DMSfKzsNv4iAVfOHkETAYFS4QNabRQ4_-pN2o 10883
modulus/models/layers/fused_silu.py sha256=yyaFOTCRKMLDAloxfcyiWgpIUifV8BEPEn2CJ26iwG8 9015
modulus/models/layers/interpolation.py sha256=mYsGqdqrVBRxvxt70m8d_vUz0yUByo1QC1ZBJgXN0s4 16601
modulus/models/layers/mlp_layers.py sha256=15gp4xKMHedMuNNB_XmkNvJS-7AEfOp0LZ2NPMmbSa8 1439
modulus/models/layers/resample_layers.py sha256=-y2qsndAct7prTNEZS_9SZUIscPom9g4VZtwtzuPdzM 7607
modulus/models/layers/siren_layers.py sha256=TI7g-A4dzA_JGPjAoJr-s9WQvyF2_HNH8TLeJfpmnpU 2558
modulus/models/layers/spectral_layers.py sha256=QwAJBap9za53qROFdGW5OGk_hWJ-LQhAOxt-67DbUD8 24271
modulus/models/layers/transformer_decoder.py sha256=iyi9CqhJLRI4fgRj4VNCDQ0p9KefWKb3VGE7xai5aOI 9265
modulus/models/layers/transformer_layers.py sha256=e2l0A698r5bq0VkqKu_j6aX6cMOL0h_Lmp4MjDj711w 24948
modulus/models/layers/weight_fact.py sha256=agAFVC8qU0nG4xx356XYnvMy16kQ30e9ELGQ5Czk-yU 3655
modulus/models/layers/weight_norm.py sha256=LJi7sIn0rY4vu8-Tz8337y_IdxlyipXxUky04echKWQ 2579
modulus/models/mesh_reduced/mesh_reduced.py sha256=8NjMHMsuipNz9pHUMx3VeEMwNjBLXxhaRBP1y7bzAuk 7456
modulus/models/mesh_reduced/temporal_model.py sha256=PvgViBMYdIuaLuuRiqz7KJT_v0tNdKXLuFF8TF9-4uU 5538
modulus/models/meshgraphnet/__init__.py sha256=ndJgYwkN4FI9p2qjADPBY_w5O2LCPXd8xRI5yiqsJ0Y 827
modulus/models/meshgraphnet/bsms_mgn.py sha256=1-eNmRIrSDqRpSbQ0RgZZA18AOCr_MHh4sXoXVZAsuU 7032
modulus/models/meshgraphnet/meshgraphnet.py sha256=8kQ4D9wS8kkKfRbRw0dYESuGZxjMc9sQ4Zkt0bBALvE 12191
modulus/models/mlp/__init__.py sha256=OudIo06xFEWs-E6Wqd53fWrTAHWFm6Oai3dAp6FoYUc 761
modulus/models/mlp/fully_connected.py sha256=ZBhrryW76z2mR6Aw2ILIBBH-OTEPxCsagZ_8i7bku14 4603
modulus/models/pangu/__init__.py sha256=6db-qybtUItGAp9x9aMmEVNMMbA7WciZDxkEt-T7osM 742
modulus/models/pangu/pangu.py sha256=IVEWf_0YemQ_jv9_XIAmrKkX5mp1k6KASGfZyLv1S90 6473
modulus/models/pix2pix/__init__.py sha256=fJia9g9QLY9FYF71YSSx2ys2_KLTxxtse-liIM4iM1k 746
modulus/models/pix2pix/pix2pix.py sha256=tywNqGO6PwyYLOv2juPIEO2a8X3Jj4M41fwlV6ZMfdE 12183
modulus/models/rnn/__init__.py sha256=PYfmrybD0WZLgklgxuskcPrTCGuNawsZm2C-MDntRNw 791
modulus/models/rnn/layers.py sha256=fIbMpkfv39kWokd7QixYT0toZSJ4OvuyBMimMzlZxlU 16242
modulus/models/rnn/rnn_one2many.py sha256=jRyQnmvE7svmtoDeUHEcO6ryxCgFPaQvGHkQlLDXypw 8354
modulus/models/rnn/rnn_seq2seq.py sha256=3M3njb70AM1ufwcAg1a45cQJl_e0FYerE5aZbw5ULZU 8669
modulus/models/srrn/__init__.py sha256=i08PYycRVB5pwxkDzAEccj-_PRoPEXdYnd9IBLsqlZY 753
modulus/models/srrn/super_res_net.py sha256=SnhulE8n4vo6zh5BKzQavoDqkeJo7kvOlSRIPqwopbM 12099
modulus/models/swinvrnn/__init__.py sha256=ssRq2VKcH59ipZpOa9FVR207yGUwSPWNpKUPgp8wiHY 747
modulus/models/swinvrnn/swinvrnn.py sha256=uBahBFU5y3umNpa5mgBfE7ZWlzJoEivIrWOC1mny5Ss 6976
modulus/models/transolver/Embedding.py sha256=AQF2FTKxGIRjeSOkBVOfavm0n1Iv9hMyJ1wZKrj_ZwM 4268
modulus/models/transolver/Physics_Attention.py sha256=2nZTAh6I4qj0hntP5HSfEDUh6OozRksT_GMBQnqel1Y 9956
modulus/models/transolver/__init__.py sha256=C75ov1B8WdGxgRrmwcCz3rE0_x69zGzuPrYHbheLiMY 1313
modulus/models/transolver/transolver.py sha256=DqeqswUghX7rTL8L1HYnD4DJPsY8Nd2lZP7jHlpVuXY 11507
modulus/models/unet/__init__.py sha256=uBFnd9xMV5gN5EuWI74Fv6_Zasewt0fSK5I2HdJV1cI 740
modulus/models/unet/unet.py sha256=2Z6AmI8fClZQAp8EH7tSWrGyF_s26t8gvwvshBujgTk 26089
modulus/models/utils/__init__.py sha256=EgQ75HzIFEzmwcNt-BXRkqA2jyNwgtenbpREB-mri78 1074
modulus/models/utils/patch_embed.py sha256=MAcpuhPtZLm_-cP6SVQOlRWmLedMI8AbrDVExcJFy8g 6961
modulus/models/utils/shift_window_mask.py sha256=t8nNnSFhGXdy5kLU39jrgM-_GCML7oDRBgs7X-OhJfw 5896
modulus/models/utils/utils.py sha256=TogXVk_4owPaDGbVsxHJoyxzoYkpTnWuHa5-G-mLA2s 5875
modulus/models/utils/weight_init.py sha256=85PPBFa9LSq_xHgsljviTNqiyxV01Mtcr7xBGhdFNhs 2947
modulus/models/vfgn/__init__.py sha256=-fvVppCTidyRBXoFfYvbj1wC2w5AL38EIhFTySTGYG0 687
modulus/models/vfgn/graph_network_modules.py sha256=E7BYNTG0d30WmvtXSgSByyF754apcl67yefhnL96hkg 39508
modulus/registry/__init__.py sha256=1-xYEpNqQGCP_xFgVr-u6c9U3BjE8_lxVAAR66DU71o 759
modulus/registry/model_registry.py sha256=--AhIT9uKmkRk6VQCEx07-3_tkEPDq7sDFxtLRq-6Wo 4625
modulus/utils/__init__.py sha256=ZdV6G6F3G3itF0NNGFX2Lcux7NZ7s1TDnrfwsn9encY 802
modulus/utils/capture.py sha256=stg2f5bGSNsD-cvzms2MjWzPQViyYvyNTBJWj8_BlOA 18390
modulus/utils/filesystem.py sha256=JKwstdSQ1tcknetwU-DbSd5sLnthKPlu2HdVba6sxB0 8941
modulus/utils/insolation.py sha256=OGTint9vQv3_Uee_FBEv6bCcbIJVQspppX3FT3jx70E 3954
modulus/utils/neighbor_list.py sha256=ImhnAnsTrhEa7pziK9d3HmJQRy56HGf0pfoD6VknQoU 5891
modulus/utils/sdf.py sha256=oT3zhMPl2FbD3pGQg1av0azT-5njW3bLWsc8QhBOp6w 4672
modulus/utils/zenith_angle.py sha256=DgXJLV4OzfkV2TwAgWccrS8sHvlJBuIMT6sKfaBVGAQ 17882
modulus/utils/corrdiff/__init__.py sha256=KbW4mOjmMUMiKlM4iZwbUu58rtjEdHBFxoQMBXkvosc 803
modulus/utils/corrdiff/utils.py sha256=IzUjbAcz4aQXJU06D9N9GNr7UDDuWNb5Y2rukd65uNU 9244
modulus/utils/generative/__init__.py sha256=b4S2alhOka7kAZEUAjENX-J33EDqcWtbdaqNpvvdVJU 1616
modulus/utils/generative/deterministic_sampler.py sha256=lrkE9bTTZJEUH1-tq1lmNIJBlw8eqEo69SqxykJBDn8 7422
modulus/utils/generative/stochastic_sampler.py sha256=pJV9ikDuxJ4THEWSBfMAc5-sIG58FGhl2augY8E4YWk 20166
modulus/utils/generative/utils.py sha256=bJXRvLRM5RiGz172D1IsTnmCRe8tE9wb53OfEP5VZpM 26471
modulus/utils/graphcast/__init__.py sha256=6RCPqCXnxF7xs_LolQ94VlK79E5t3NNqmBsi_d7JwOU 716
modulus/utils/graphcast/data_utils.py sha256=l4JOKRws7Dv3yCdKOEMnbbFVH9XuT7y_QO_KespkGIU 4299
modulus/utils/graphcast/graph.py sha256=k3S7RiHEAw_nqDRjucVsOgx-vCw_d2_SyBme4484vDQ 9785
modulus/utils/graphcast/graph_utils.py sha256=rJYyrNGGA6y1xp-qZZKQwiOvq7E67aujg8ZWVF1RdqY 12923
modulus/utils/graphcast/icosahedral_mesh.py sha256=gtuaaTV_bY3_H_12OKMAOksbMLC2qN3LhBwB19_OoqU 11757
modulus/utils/graphcast/loss.py sha256=bYkxNJzuvzFCBCPZnHJmaMEUxmc7QGyFbV49Hk-olYc 8090
modulus/utils/mesh/__init__.py sha256=4a7zPUMFAKc2raIhkozAWNZxnS4FRxi8OnViu0kDiHo 875
modulus/utils/mesh/combine_vtp_files.py sha256=VBZlO3JDQmMCb2ZFuw5v4Jy8cetPl4Y01UYiI7xaLIw 1590
modulus/utils/mesh/convert_file_formats.py sha256=afXRlBwld_hziWxWY7bveVE_xS9zjx3d9HPCm_mbuD0 3304
modulus/utils/mesh/generate_stl.py sha256=wO9C0O9zwrc2_l0gFW7tkl_uxfXTyV16yPvwZKkH_qw 2871
nvidia_modulus-0.9.0.dist-info/LICENSE.txt sha256=-ATP1Ss7rP0Wy56Awa9VzNl6DEEVn8BkUte4tEQyFJ8 11314
nvidia_modulus-0.9.0.dist-info/METADATA sha256=GRbev9gdu8ROKczqQcoq4X9NAiZ9saiq8tQZqV_tMps 21953
nvidia_modulus-0.9.0.dist-info/WHEEL sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY 91
nvidia_modulus-0.9.0.dist-info/entry_points.txt sha256=DbY22D2h1ZUYLxe8Vlsgn1YJqZgKXV-b-A-rNu6FOSM 601
nvidia_modulus-0.9.0.dist-info/top_level.txt sha256=Z4xMRahCKo9D7ZQXDoTUqVdssItN7HaeXe18Jx7VLkA 8
nvidia_modulus-0.9.0.dist-info/RECORD

top_level.txt

modulus

entry_points.txt

AFNO = modulus.models.afno:AFNO
DLWP = modulus.models.dlwp:DLWP
EDMPrecondSR = modulus.models.diffusion:EDMPrecondSR
FNO = modulus.models.fno:FNO
Fengwu = modulus.models.fengwu:Fengwu
FullyConnected = modulus.models.mlp:FullyConnected
GraphCastNet = modulus.models.graphcast:GraphCastNet
MeshGraphNet = modulus.models.meshgraphnet:MeshGraphNet
One2ManyRNN = modulus.models.rnn:One2ManyRNN
Pangu = modulus.models.pangu:Pangu
Pix2Pix = modulus.models.pix2pix:Pix2Pix
SRResNet = modulus.models.srrn:SRResNet
SwinRNN = modulus.models.swinvrnn:SwinRNN
UNet = modulus.models.diffusion:UNet