neuraloperator

View on PyPIReverse Dependencies (1)

1.0.2 neuraloperator-1.0.2-py3-none-any.whl

Wheel Details

Project: neuraloperator
Version: 1.0.2
Filename: neuraloperator-1.0.2-py3-none-any.whl
Download: [link]
Size: 186926
MD5: 2f9d4deeaa027aa9b2be688581387058
SHA256: 9a416ac81cb027d95c2351862c569053bef849f480b8ffb07fe78338523d4a5e
Uploaded: 2024-12-30 21:00:50 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: neuraloperator
Version: 1.0.2
Summary: NeuralOperator: Learning in Infinite Dimensions
Author-Email: "Jean Kossaifi <jean.kossaifi[at]gmail.com>, David Pitt <david.h.pitt[at]gmail.com>, Nik Kovachki, Zongyi Li" <david.h.pitt[at]gmail.com>
License: MIT
Requires-Python: >=3.9
Requires-Dist: wandb
Requires-Dist: ruamel-yaml
Requires-Dist: configmypy
Requires-Dist: tensorly
Requires-Dist: tensorly-torch
Requires-Dist: torch-harmonics (==0.7.3)
Requires-Dist: matplotlib
Requires-Dist: numpy (>=1.25)
Requires-Dist: opt-einsum
Requires-Dist: h5py
Requires-Dist: zarr
Requires-Dist: pytest; extra == "dev"
Requires-Dist: flaky; extra == "dev"
Requires-Dist: black; extra == "dev"
Requires-Dist: matplotlib; extra == "doc"
Requires-Dist: myst-nb; extra == "doc"
Requires-Dist: numpydoc; extra == "doc"
Requires-Dist: numpy (>=1.25); extra == "doc"
Requires-Dist: sphinx; extra == "doc"
Requires-Dist: sphinx-gallery; extra == "doc"
Requires-Dist: tensorly_sphinx_theme; extra == "doc"
Requires-Dist: torchtnt; extra == "doc"
Requires-Dist: pytest; extra == "all"
Requires-Dist: flaky; extra == "all"
Requires-Dist: black; extra == "all"
Requires-Dist: matplotlib; extra == "all"
Requires-Dist: myst-nb; extra == "all"
Requires-Dist: numpydoc; extra == "all"
Requires-Dist: sphinx; extra == "all"
Requires-Dist: sphinx-gallery; extra == "all"
Requires-Dist: tensorly_sphinx_theme; extra == "all"
Requires-Dist: torchtnt; extra == "all"
Requires-Dist: numpy (>=1.25); extra == "all"
Requires-Dist: scipy; extra == "all"
Provides-Extra: dev
Provides-Extra: doc
Provides-Extra: all
Description-Content-Type: text/x-rst
License-File: LICENSE
[Description omitted; length: 5892 characters]

WHEEL

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

RECORD

Path Digest Size
neuralop/__init__.py sha256=A6ZKhtee-FELHAOngIOEHrvVsZATEklbPdXvcZQDqtE 265
neuralop/utils.py sha256=b7JwCVUSQzSooFpZxvRjuc2vaG1NBS8O3Hk7-7hJTXY 6358
neuralop/data/__init__.py sha256=owdlNle4Ezhvp_mtJoVcbVG596nEGMV0XT5CQI7lo6E 223
neuralop/data/datasets/__init__.py sha256=NIf1vqvujNiArspZiUKZPlgNU7lhzGDI7OCypiSON08 490
neuralop/data/datasets/burgers.py sha256=SA2t7TCo63ea46CeFsANL4bYVvFRgbFGSHbiz3X0H70 4053
neuralop/data/datasets/car_cfd_dataset.py sha256=FKEth5t4Auf94L8DcWWMDLzskmlZaBUhs43xv5GmcgM 2963
neuralop/data/datasets/darcy.py sha256=0_7tCHYXHFxbM26xBn_rfkPrmWhHWphfRb_MfGlky8w 8684
neuralop/data/datasets/dict_dataset.py sha256=KUZ1Ypy-GQP7dW5GQ5F-2z2BEWvh6XQdQ8bTmzg04YA 956
neuralop/data/datasets/hdf5_dataset.py sha256=UlHLHx5aI_lMKLoLYZhpwPU9Zbpq4uHaO52amX5G_eM 2195
neuralop/data/datasets/mesh_datamodule.py sha256=tjbrl_slLfJ0CXGpcqt84dV_HaAV4h9twDIVtqeyua8 11213
neuralop/data/datasets/navier_stokes.py sha256=qb2mRWl47SI_EiybWrDNvgzgtALfEYQ9TEnfb2uhDFc 7013
neuralop/data/datasets/pt_dataset.py sha256=mgXmsEFxmpsU0KOywcemmXMvOM5N3ooxwMaoZ-QZB9c 7894
neuralop/data/datasets/spherical_swe.py sha256=8w5pWWH0kabmu1Q19BsfJL3v0BExS4Kz-YzL_21Dmzw 3678
neuralop/data/datasets/tensor_dataset.py sha256=6iqa8QdahB3m3wFUiWR604ydDeMvShpkyXdJKlqtGHg 1624
neuralop/data/datasets/web_utils.py sha256=FfTtLoScDHwAacr67AFsiu4IasEAIeCM2XsDJAReCnM 5554
neuralop/data/datasets/zarr_dataset.py sha256=iDfzIrX0gl6saVeeQ_UruPQE3IZ1EgaS5s35xcyJVxg 3118
neuralop/data/datasets/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/data/datasets/tests/test_ptdatasets.py sha256=Y8rv3Ro56-di2oppTpg3mJsONjlBj64Wo3O1QgRgttc 2409
neuralop/data/transforms/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/data/transforms/base_transforms.py sha256=DhO3fZ7obvBg4UT9oaObVL1v_525vRMp01fWQfsJzxU 4008
neuralop/data/transforms/data_processors.py sha256=sRqpUyh69dzdCUjVAugaKoeESDHkes_Zf0JmBbufNw8 14444
neuralop/data/transforms/normalizers.py sha256=x60rZPNTgWvUDpb3DjDR1M0kgB8NGJ2ipld6zTnu_oU 9685
neuralop/data/transforms/patching_transforms.py sha256=6GY-gBpRQXbl32QfhKVEqRRukHRuoA4lSWGaqffX-tg 3824
neuralop/data/transforms/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/data/transforms/tests/test_data_processor.py sha256=nNAT3Rhwn0rozrVY_B_qmmkjRpQaqRv7_0KoI-QTLR0 4339
neuralop/data/transforms/tests/test_unit_gaussian_normalizer.py sha256=Hc1XC7Zw9NSn8b0QyIBjQvcFdaGIaiWt6xDE52xFo4s 2290
neuralop/layers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/layers/attention_kernel_integral.py sha256=MCuGNkG_7dL46E2jkWZG4Kdnagqvxiq0GlL_OHaWVRM 11101
neuralop/layers/base_spectral_conv.py sha256=nyTiKCoaqFmLhruoSG6BIOIdpmw0E_0A8Xu1bvX6d88 983
neuralop/layers/channel_mlp.py sha256=8hAfWg-PtqtFB8E1VgX9X1mxG3b76f5ZH5UuXpOhyAU 3738
neuralop/layers/coda_blocks.py sha256=aY-Dqf11zwT51T-qqqYd6yooalVuj5QFViklQAv-ivw 16697
neuralop/layers/complex.py sha256=KIcJGsMVEukOWveMyfsq9UDZ7RUnSL1X2oaR530FHlo 1872
neuralop/layers/differential_conv.py sha256=K8GmN30P1fgcWn1jagauhSdtGSpzyxyXE4O2BGOb-_o 3301
neuralop/layers/discrete_continuous_convolution.py sha256=SGHXpK0wz4qDcVZgV13MLjxmQNHaK1JYnLdg4iiX6fs 39191
neuralop/layers/einsum_utils.py sha256=i3va6xxAiRsIacMCvUJ5uDPZmHQmFgvU0yL41XD0Qb0 3086
neuralop/layers/embeddings.py sha256=pMl8lEOoUir312QkspcwNr1IQSZM9Ne4AMB7F7hwuFc 13655
neuralop/layers/fno_block.py sha256=wW6Cd_CQ_q3_ppqycD2_veilBi3mhGVZYXruXoaxRtk 14097
neuralop/layers/fourier_continuation.py sha256=_zO68k98jbcMbr1Go-KFJYDz2F4_w2naBR9KwsAzmPI 2200
neuralop/layers/gno_block.py sha256=e_FSuPEmyqXubhGjqNi1NQPUmuV00-OPp-1uq5rZkUI 10378
neuralop/layers/integral_transform.py sha256=VkFJwSE4gWdK_pT0vr8eFmZsxBwzA2GsLlVEIfgH1YA 7842
neuralop/layers/legacy_spectral_convolution.py sha256=J8nFs-zrHFLufY0eRcHxOPb9S3GQRO4LQk-1CKvQ3t8 25859
neuralop/layers/local_fno_block.py sha256=7YqTHSLDNwG_6YpcTAvuZ7Q1_Lcc9UH2-KqMyXJCUgk 20370
neuralop/layers/neighbor_search.py sha256=iwnEwwmO0BDRrrkhDTSEBBsdu4LGRXqnnp77t2zEdZ0 4018
neuralop/layers/normalization_layers.py sha256=hd-dYGf_ZPGgm-DpskWj3pRwoVvd4C8DlciXwofO6Pc 1393
neuralop/layers/padding.py sha256=WS-i9eQwnjRn0EN1rEN2RNGq3_FIUgZpMcE1FVDCVbs 5508
neuralop/layers/resample.py sha256=wwwioHNxz9-iJn7CToA3ilZPODyL8YJKDYaL1phnjnw 3341
neuralop/layers/segment_csr.py sha256=w7Q59JR6EOwZ_0ErJQGS8dlnWpaSqKhfNa-opaTrj9c 3070
neuralop/layers/skip_connections.py sha256=1bh0vl5GDoJoA5HaggfuBHSvUvP9t1YXtW__Xmc5oE8 4298
neuralop/layers/spectral_convolution.py sha256=sRTZ7FF6iVFpM0lzhK7j1qXcTdhu5MxzWKxI2TuA48E 24639
neuralop/layers/spherical_convolution.py sha256=3PhxNvzmFNR8ggg07JGg5t3AfdyByU3R-H2KzkGVNT0 15649
neuralop/layers/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/layers/tests/test_attention_layer.py sha256=zzGr3h-8Fv_UVAnJKzvcLTJcjeWjSnat0l7aIMLGZzU 1023
neuralop/layers/tests/test_coda_blocks.py sha256=tjvffgNTU28v1oODNfAZMXfkH5f3RBKwGAW6YjYHLUI 1409
neuralop/layers/tests/test_disco_conv.py sha256=NozXogWhlUAwcs0L5QI85Ae0WqAjJybRSeFBGa_UnxA 2719
neuralop/layers/tests/test_finite_difference_conv.py sha256=TuJTnZsew52BiZXGpXKpyq8bD4GfRNi0sRRVDvc53jQ 2306
neuralop/layers/tests/test_fno_block.py sha256=2xtSVLagy5sXQfmJrjK3C_orbW68sVR3nj8W9wmaCuo 3413
neuralop/layers/tests/test_gno_block.py sha256=Qp1RUUyRM-YmpsGUnMTC8MkrhbL6F4AcbbJm57XnXVQ 2788
neuralop/layers/tests/test_grid_embeddings.py sha256=e0HH3EFmCFmffC7zMTiXz8VxbZBfytnZnFYm5ZnAlko 1476
neuralop/layers/tests/test_legacy_spectral_convolution.py sha256=g7iHYf1gfyl2NNbDTjSpZPIbViSLBVR58poYOn_Z44Y 6374
neuralop/layers/tests/test_local_fno_block.py sha256=T6e8r0tSjfEK7rYb7v51Z9bwejiXNAZdSmwWShbHUPA 2872
neuralop/layers/tests/test_neighbor_search.py sha256=Fd0U8MJG8dflzy-bMmHoBMZ29CprxcoE63Ktfgfrkx0 1311
neuralop/layers/tests/test_padding.py sha256=NEKfWe7ZZxjA-LDR1bgYmwqd7CIPDl0edpl1DMEbFVk 532
neuralop/layers/tests/test_resample.py sha256=Dmn7-EukXX99HD-kgZOkPBxDpZ6LBhMkioPhO8RtEWY 503
neuralop/layers/tests/test_segment_csr.py sha256=dccfeLPvKGdkaWTvoqDteVmSL_Vp4ruVPIgcxiM0Y3c 1591
neuralop/layers/tests/test_sin_embeddings.py sha256=EDyWJgwey6xXZRmWFWLWpY5WjVVZg4PjAsgA5Q6-EKs 4021
neuralop/layers/tests/test_spectral_convolution.py sha256=5Tziqv3Ovt2lrkL0uDMBTefWdxIH2ICWOiqX04af_IE 6687
neuralop/layers/tests/test_spherical_convolution.py sha256=BAiqjOPnY--X16Kbo7d6jyH67uqKorL3vmKTsV6mfBQ 2972
neuralop/losses/__init__.py sha256=y6hvj55ytlL5Jocrcsj74uH-3rA_DIJi0pkh9spr6kA 132
neuralop/losses/data_losses.py sha256=7nEi7sz2fblE1QDS13j6J2-QqTR3U5WEpSh-b_qpdoM 15554
neuralop/losses/equation_losses.py sha256=0c2Tx3dydLz7I-U4j0OXb9bCgtV5NEc7nNVOXmzYuOU 2056
neuralop/losses/finite_diff.py sha256=SBcWXm_IVTI8dk-JJToj9utI2EibVo2gofoYg_rBKYk 3869
neuralop/losses/meta_losses.py sha256=GFlCy5W5Phmdy9v4Gra-etPm8GvLx6TTWoBl1KQMkT0 3024
neuralop/losses/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/losses/tests/test_losses.py sha256=y8DnytqXs_9QCVJxoJM3n6c2Rx1nNvtq_grHdm5NADQ 3712
neuralop/models/__init__.py sha256=VrD6S4-lk9todmByTfjQ6zDgA4BuR8hx03j3zk-kkwk 372
neuralop/models/base_model.py sha256=QIwcd6u3oEVfqUb7WIV-94mLa7Cu9yGtOckJafy2iOA 9492
neuralop/models/fno.py sha256=q-70xYdYt97U5v3plcDPVSi0Zq0Mj-XS-ovbUo9ecdw 24432
neuralop/models/fnogno.py sha256=HggK4b0PTvxNkc1HXeRTsZe1H96nXCvc-ytEAvx4q3E 15163
neuralop/models/gino.py sha256=6EhhrAA-QnLKf8DQsXnv8r_7wNOhnGQgq08BwMFdU48 20493
neuralop/models/local_fno.py sha256=PsQpl53GnRl0KdQe3DDryyLdzLxWkb8ANlyDX8MQReU 18312
neuralop/models/sfno.py sha256=TZejyAFwo2sYNnbnOBZM1GMdjot1N2yeLXubtvFndCk 582
neuralop/models/uno.py sha256=1BzM2FmDvLq0visDx41HVvaM-yfLoyWUONmp8rDPEg4 14996
neuralop/models/uqno.py sha256=S5ixDh9_tnwPM-DUQgrST0C765D2qk2Y8TqSkOzD-7o 2025
neuralop/models/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/models/tests/test_fno.py sha256=oniF56QK3po1LR0LMuH0ARKPtjR5jlgjXGYbocZEZHY 3659
neuralop/models/tests/test_fnogno.py sha256=F4ZPgJ92DRsxCcuAVMJmlowAAUoDA8HNndPbLb5gjoI 2488
neuralop/models/tests/test_gino.py sha256=8WpJMG0A1v9YHVgi5GZ11MAxcOEzix-eXUwIFws_g3o 3637
neuralop/models/tests/test_local_fno.py sha256=j1gw-I5tkK2X8NnQ6HA95VVPApXLlrcZ7964Qn0u1-E 3294
neuralop/models/tests/test_uno.py sha256=D1w7gjhbLOfSz3gkmjtRzrOSMh52yyNaei1j9tCzIvE 2044
neuralop/models/tests/test_uqno.py sha256=9rg4vuhTvm4yJYUburonkk7g9Wgw9jVVI9fMYkvohd8 945
neuralop/mpu/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/mpu/comm.py sha256=f_rV3mtQdiwAyaMDaloahSO04HiywgCXL-eHL1xjymE 6086
neuralop/mpu/helpers.py sha256=LhtywmZp4lvzIxh09NVVUoWSv3B51sLzuvGYzBqO1Nw 5179
neuralop/mpu/mappings.py sha256=RC4mbtB1jetMvyz5kwDd2fdBqgl_Lr2NJbfc7_wEFio 3502
neuralop/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/tests/test_model_from_config.py sha256=2xkFwzyq7cz6ulKcXIPmpxnWrtkCqT33ACHKgFSmBcw 1071
neuralop/tests/test_utils.py sha256=CfJ0wFt7jLiuRaMkU2OSpFG9MnJf45CG3YB79kPlXGk 4139
neuralop/training/__init__.py sha256=96iVq3llKN-uqsBfUtCUctT_NqsnpWYmz8PbYFjL-eg 200
neuralop/training/adamw.py sha256=WfCwIT9tfgvxD8HAzyoVne1HQBwmf_dn9eMqq9zNNIY 5204
neuralop/training/incremental.py sha256=yW2uo3kyZ80_IGuNqTehUQ87uoBjtw0Myzws_XfDcEQ 7040
neuralop/training/patching.py sha256=jkYinIm5o1H8S2bAuoSdb16bXXK68_PeLnnfjkyZ0iY 12147
neuralop/training/torch_setup.py sha256=txJYRZp-dj-heRKcQOkhCgnYpGdY78R_LzzW3Kx9Ysk 3488
neuralop/training/trainer.py sha256=PjpV5TbosS9Zk2dUqxcamvdC8m56lL8q2w6wHjD6MVw 22741
neuralop/training/training_state.py sha256=qLW-JT5MO7H0Z_4rqQllK9hR6n2XDulDqrTGocLxy9A 5864
neuralop/training/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neuralop/training/tests/test_optim.py sha256=UG5BuoWr-RYVW61TR1bzJv2Dp6-XNB1NC3conyzp24w 879
neuralop/training/tests/test_patching.py sha256=4F3M5fibL7JDEUITnTr5oFHDLFtRka9DxQTH8DWGKCY 2604
neuralop/training/tests/test_trainer.py sha256=aRzX-EW4qrcWJrbVXNqccA9Pwfi4vdt0hb17u5KPFIY 7846
neuraloperator-1.0.2.dist-info/LICENSE sha256=XzNbBgOs-x0YC_MRdJLiB2ViB1zJtjU6O5TV1bwD7zM 1082
neuraloperator-1.0.2.dist-info/METADATA sha256=Q3DgiFZJdAX7yUwoLXDKCmW62tHxIdMWmWBaoS31Qpw 7538
neuraloperator-1.0.2.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
neuraloperator-1.0.2.dist-info/top_level.txt sha256=nenaXhqR-FbiM4ScZC39jTBzlVw3p50MlEGnRw662Ws 9
neuraloperator-1.0.2.dist-info/RECORD

top_level.txt

neuralop