neojax/__init__.py,sha256=BXM3_M8WZSUvCr6OEUbp8aDim1rBDNnBMX1X2d8yEwQ,83
neojax/benchmark/__init__.py,sha256=lFhSLQeD8b73txelk-0wo4GtbZOQSiXrwjGtGWAskJk,730
neojax/benchmark/config.py,sha256=TEFNv8BG-NDDicaBq8OVXP6OP5ft6HUTcBvFbdj0qH4,6584
neojax/benchmark/factories.py,sha256=z3j70X-vI9aUoL1Fp7RuN_uET7kFgD_AoRhZsIaMHXo,4498
neojax/benchmark/runner.py,sha256=KF3cw6Y3d4GTTef0mDTzRMYlkTTyesitBQTBZiwQ7nI,9758
neojax/benchmark/evaluators/__init__.py,sha256=Avh_h7yb15Jjcp6E_Wm5N_SY2HuHdNhVRt2wdtl-K8E,567
neojax/benchmark/evaluators/base_evaluator.py,sha256=t6dm9c3KOGoju128SI-A_fF-HI18UgOC3Bt9Fs0SMLg,9325
neojax/benchmark/evaluators/steady_state_eval.py,sha256=noObqSUki5gn74srLNOh94BibETWVvLUtLnK27w80uc,3027
neojax/benchmark/evaluators/time_dependent_eval.py,sha256=ny5yrhmusFEfaWgt2RldFdaYF3cwJ-fLKQNl3cnoWUE,7057
neojax/data/__init__.py,sha256=lWivO__FYSC82ZHivcNAhZPyvG3SJqzxlaRYUjvz-vk,506
neojax/data/processor.py,sha256=24MrnQWhGHg4x7M7KICcX5OrrbUDGITGTuQ7y6G_gYo,5823
neojax/data/scales.py,sha256=JOUO2LbajBZ0LXZbBFXwW557R9PciKGP9h2eXvN2XUk,4884
neojax/data/types.py,sha256=kvKkCv22K6Pf8urCW1GNtpt1snQVNtlKC2ddEacr6-s,1190
neojax/data/bundles/__init__.py,sha256=89afchgmjKIq7PjFZ3XlTsC5oWYz6lcvq6hNCpLeCYM,116
neojax/data/bundles/data_bundle.py,sha256=lq128hzh3tDFdNw_dFZhQ5Q1rNXFqNJoNByGqRa_gdA,3457
neojax/data/datasets/__init__.py,sha256=d1ZsJy8nZeNNE7CKbBNPsFUjge48PbA66jKE-NA6Dps,323
neojax/data/datasets/base_dataset.py,sha256=W-mHNu728B7m95ji82gC4JkLNoNWA72-gHDz_yRpEBY,749
neojax/data/datasets/bundle_dataset.py,sha256=SPCThA57NJiBoFDYjNeprnQnPaE6i5tXJgDOGackXG0,12570
neojax/data/datasets/raw_dataset.py,sha256=ZfH7IWb2Pf08zCpNSbL2r01v0yUQ3Js3zeuzQgcPEqY,7263
neojax/data/datasets/utils.py,sha256=avnDcS0-k4Ubyt6f7OLlwWuLQW8Q6mmNou59-uze9Pg,19883
neojax/data/download/__init__.py,sha256=fgrJl33ntfQgI-wdERfCMt8zRmJ3Kwmoq081uxu0LFw,229
neojax/data/download/base_downloader.py,sha256=9oQxeqKF3PTOWUyHZl7Jd97VjHxqVGeFkp9Lo03Y_us,781
neojax/data/download/dataset.py,sha256=pK3P0XLKNDnH2pWUkgVIdshSk0GBtZZASR3RmY4ph74,2091
neojax/data/download/dataverse_downloader.py,sha256=Er0xbcxy6gDkCauCS8umWWG0d34b-o5Q6b3llG5F2jo,4894
neojax/data/download/http_downloader.py,sha256=ufP6Alkshysj5uuOH8HU8V93E77BUPVCacnkSid9AqI,11935
neojax/data/download/huggingface_downloader.py,sha256=dNfJxJ79pwbcurLizoa2pfeYbeJZ2qc_odyMrA0BwG8,4545
neojax/data/download/registry.py,sha256=NaNOOvTPyZoEL3kSw41lrp7STE4xCvELojZkINNbxCk,12272
neojax/data/download/zenodo_downloader.py,sha256=hNcorbj63ASmnaEQrJQez3aKcjoiesBHfNccMTrSbC8,4200
neojax/data/generation/__init__.py,sha256=hAWusL3bx3NSufOrSrk6-f6wP7wjJ4ZvyPU1WRuFZNw,151
neojax/data/generation/burgers_1d.py,sha256=lKWgHlD8KmUI-fZm0VzMWyWc5r7A-t7xiGzu5PI0RSg,8151
neojax/data/normalizers/__init__.py,sha256=G9ZGvW6vwPk61i9MDrfWxzHejqPpIx7hTw3UU54sLtQ,651
neojax/data/normalizers/base_normalizer.py,sha256=uCMifQKEqxQ6wwHWsnI22Qiy7ZcatIMMsTHo0ARbHv8,1808
neojax/data/normalizers/composed_normalizer.py,sha256=JtARD54ErUCmV8uIX4EpiwwELjxsE8_JE3jXlTCVcmQ,4498
neojax/data/normalizers/min_max_normalizer.py,sha256=I2PHIbIH8uqtNspOlSAeuPDESFz5A3vUb-D5wzrH3fQ,6030
neojax/data/normalizers/physics_normalizer.py,sha256=WxgFEeNkK8orWiSVGX5iQe4fTIfX3fSXxLKEvqeMFYQ,3533
neojax/data/normalizers/robust_normalizer.py,sha256=FI5BauPa_W87UhXkq1kjlkRSPoeafRuRzHwEJ1_UXtU,3771
neojax/data/normalizers/unit_gaussian_normalizer.py,sha256=ypWAS6tIdKclLNVlvTS1fl6R1UWrb6VxvD4dX1Qmo9Y,3098
neojax/data/schemas/__init__.py,sha256=T85Cjhjrj8aVPZkhAAt2PQzNv-gXdVtEj_JKefcAlMk,1787
neojax/data/schemas/base_schema.py,sha256=uE3WJ2pOf3ajRRAxt0RqS4fCVNFUL8mOTyqkyWBhKyI,1225
neojax/data/schemas/bundle_reconstruct_schema.py,sha256=ifXXMfpjBYGe7zRF0t5GqREgurykFFslFO8QNsXxu7c,3353
neojax/data/schemas/composed_schema.py,sha256=4rCHcr5HYRUAzvi2-K9AZkw_CMm9FtpgqzMvxnMDiEw,3634
neojax/data/schemas/concatenate_coords_schema.py,sha256=e_d03tkWPs0UBvAI3N8SXfOKX0WUj0KHgF7lSyRofIo,4028
neojax/data/schemas/concatenate_params_schema.py,sha256=hClKkYfocvu8H_dQ8wmLjD2dnvzmEJaRNV-9uLXz-bM,4202
neojax/data/schemas/flatten_time_schema.py,sha256=9giyo1X8H1rVUyMIoEtzZEPtGPYL9HZ1FNYrV11qWik,2494
neojax/data/schemas/flatten_to_points_schema.py,sha256=6GEGOCgTl2L0hEmkpgkL7tJFL7gMF9SP-4r2ZhfThGI,3816
neojax/data/schemas/graph_tuple_schema.py,sha256=QArCQE-UGInTvuUcdkX_phnqn8xBhTbnZGNvmb16YgQ,4073
neojax/data/schemas/identity_schema.py,sha256=g_S5-7NIO1SL6-LVLYw1SYISlNxxoIedbfb8yV3GK8w,958
neojax/data/schemas/mesh_schema.py,sha256=45Rtv7CBpRayhvul0sedJkQFSSCqb05xSvmTaz6fpac,3977
neojax/data/schemas/reshape_points_to_grid_schema.py,sha256=7L0rtwJAEXWwTirZe-m6QeBL_f9ipNbufkq8TqrxsgA,1992
neojax/data/schemas/time_to_stationary_schema.py,sha256=qXAiZ-vAn6xRKA7vobVUyhOBuSnSTSkwBgJ_wsqXuPU,5327
neojax/metrics/__init__.py,sha256=M-B525OBPzww85P14SHG2-xyYhfpoCg5jxlWvZhhOEU,1142
neojax/metrics/base_metric.py,sha256=5AmF9PUWWbxAhLMdbS3LHVamRCqUb24qPSjgWv-aGdc,4250
neojax/metrics/composed_metric.py,sha256=DUyPOuUEmGY5TaRGi7oEBf_ihzZ3VSx94hJ5exiM1DU,6861
neojax/metrics/lp_metrics.py,sha256=MCLtrg1BZSOqR5inu1zuAjOcOw-GhejYXDMtearFFnw,7246
neojax/metrics/physical_metrics.py,sha256=La3iHCDcqbSZUaCFCW4ceYAGTgjGpdW0dFIfHtQG80Y,13345
neojax/metrics/regression_metrics.py,sha256=skO4jrSZIBDbpxw-e9KCmky7vX7mxOHsVfzjRfsVbO4,8938
neojax/metrics/sobolev_metrics.py,sha256=7I73pa7bGzcYaf1S7qEAxIFPETon59dR9tAjyHp4JKU,20746
neojax/metrics/utils.py,sha256=lQ8F5m01GQve0oXZBmkNGEGQqqvcosyoGq3ospj5iyQ,4124
neojax/models/__init__.py,sha256=l9q7m00z0jbh9cTdm0D83A5BqqKU_Ad_39sALRDJEe8,789
neojax/models/baseno.py,sha256=ScWVymKOKOFOAFo2F7FadTWBDct5LtZUNsOekQK_7V8,10985
neojax/models/cno.py,sha256=evR5euO0Zq3ndaW5hHAt2FTNtEL7GwUR4CuE59YZbTE,20038
neojax/models/deeponet.py,sha256=hpqjI0yA_T3cWyGUg389r8QSfh0AP4wdQb8NXviP7a0,6616
neojax/models/fno.py,sha256=PddtcKhkkW7XrRQCHGVbu0vVcfiXQ7hjdMJRuC1F9Kk,17635
neojax/models/geo_fno.py,sha256=pizQaGDrCMIHyDE06o2LAIvYP2VT5Gl84FrNxKPyggM,15759
neojax/models/gino.py,sha256=nOLjr7B3E2-Duoqw_VB7I_5rzuOynBUQQcvcVFMfY6o,28241
neojax/models/mgno.py,sha256=9hqB-HBdrDT275DRxb_twDTWE79it_4jKx9DkaaT0vw,19525
neojax/models/rno.py,sha256=wdIoqKFeHWkWKaOICdJsjwNM0SMFNdkeUzRJXUhPRwg,10133
neojax/models/tfno.py,sha256=SRiKJejjc_v6FnQYMOfQPtK7bkPYZ7pxrTLlT8mIDo0,11744
neojax/models/transolver.py,sha256=Z0pzcneG7VBQ21WmQJhh_wiOdl2I7zCj1FKeQcj85PE,16520
neojax/models/uno.py,sha256=L9_NCWdrHMA4uYmBWK9f6FLVWy52iRy1VK4JlYqAhno,21413
neojax/models/utils.py,sha256=qS6lia8VffRvFUGgGgoTjfxjzEef5Sq69obtZADV6rg,1728
neojax/nn/__init__.py,sha256=NTxQ7ULo3bAnvsK7rH-_B5FNNCSRoteyqKjMBO4FwQ4,3356
neojax/nn/cno_blocks.py,sha256=9cAUxm4CgXnivuO7SPRBQuiczBvmRuQ_EVq2BblElgk,20356
neojax/nn/domain_padding.py,sha256=swnKUC_hUXUeJ-jrLEskyzjCUvQUBSrD7yQfciPZSuA,7410
neojax/nn/filtered_lrelu.py,sha256=JgH_ez-vKL_c5eHi659RmqGk-cosVQNbxc4lHsXEzVU,11375
neojax/nn/fno_blocks.py,sha256=oTALiimVoR097KL6IRo6NidmWCu7sO33uWkGwdyv1OM,30256
neojax/nn/geo_map.py,sha256=J-xTQTpsuVPjtLV0XxU-qcavrg2w6aAI8qkizgYGaKo,5248
neojax/nn/geo_spectral_conv.py,sha256=EAB4ROZlm_FPtDTQlnzWxcUUPrj7o44PUeKk80Bo7B0,6989
neojax/nn/gno_block.py,sha256=UJL4a4izplQWsJm-8yKhA2-1i6Q86gl3mbYEwjjBLss,11372
neojax/nn/integral_transform.py,sha256=BR-VBXMGeEZ4MOG27bysYKor0Ly3xO3xbdSimh4Dgzw,10249
neojax/nn/kaiser_fir_filter.py,sha256=wjwM1adnfk85PZ2WIBAZs6pdgON1OV91z9TWUIdC5J8,4980
neojax/nn/mg_conv.py,sha256=Su3oxpN1pyQIcn94ANfCwX0mwMablg2L2gplpdTtLC0,21533
neojax/nn/mg_iter.py,sha256=PYuAG3quBCjfKXb32qHnB5zuF1eK5-XR-s5JX-ZdWEg,1825
neojax/nn/mg_restrict.py,sha256=5Z4IHpyMUgs5dC_1JaQyozFi0Vo4vF7kbrOCUBvYe50,2033
neojax/nn/mollifier_functions.py,sha256=o9p1WK_ZmadQnKz9RpNSsnR7Pr-KFQGbNCZId5BUQuw,4434
neojax/nn/neighbor_search.py,sha256=5u5JglsAU8yAY6WHMqqVVEJ3WNMsYI0GEePHEeGQnSQ,4591
neojax/nn/normalization.py,sha256=NQC441N_DzeQStRUCLWyWXd_NFMe7VKYexZmzRWSAZs,4860
neojax/nn/physics_attention.py,sha256=JEUqryLDsnDLxBo2DSH2pH1Iw-Spf1CnyF7slnBH-iA,12729
neojax/nn/pointwise_mlp.py,sha256=0ZBbAw0Zo42WeDcGEXashwxMxNAsf5yeE6uxGtWyt6I,6078
neojax/nn/positional_embedding.py,sha256=TVBO6_ukjfYdhABrWQy1FhTFFVSwdO5y7gn7k5Wc44A,13366
neojax/nn/resample.py,sha256=RG5sDZosc7yfHnhqUCgCB9GiSfZaMylj51eCiOEZd58,7955
neojax/nn/skip_connections.py,sha256=ZkX1F6XR4myEmizA2cJUbM2KHVDg6youiaFp-C9S2vQ,5140
neojax/nn/spectral_conv.py,sha256=t5Trw40RGzvcVGQZ_hpFVjT733Wxq3H09ieiUNWQU04,15493
neojax/nn/transolver_block.py,sha256=9sgGQ6dYv5e1Sljo-zg6nNHmACTIpaBKk6di9FfUWxo,4096
neojax/tensor/__init__.py,sha256=QXZvjTZej26GxYYzkWNSw-4AzEPz_-nTrycHVCFvPDQ,418
neojax/tensor/base_tensor.py,sha256=wHfps6hYgSMFeX4eYRRmkHm8_A0AW4vRSjKuZAHToX4,1028
neojax/tensor/cp_tensor.py,sha256=lsQZU031mfY2KJ1ahxICOIvtxl65kt55BwgPoUs3Bf0,11046
neojax/tensor/dense_tensor.py,sha256=K91KYuysoi8-Cm2rj3FUDghUvW32nLnRyV3du9cYG8I,4315
neojax/tensor/tt_tensor.py,sha256=IwbAZvwOUj7gLhVc6waLyxOUitr_zzaevbadxn-nsGc,11843
neojax/tensor/tucker_tensor.py,sha256=3lI50fwAK7zTpPE93r-E3DfOTR8rqMa34fOIaeEH2m0,11350
neojax/training/__init__.py,sha256=VGFRiIAQSubGmeemw6UTsBKmlJRlaOlf5ESPqpdkzIY,376
neojax/training/diagnostics.py,sha256=rZ2BoLLk1p2WZfuvCM7x7rjBCBDXl_6KJ3HA4dlp4rA,6619
neojax/training/trainer.py,sha256=yd0cfDSis579AFh7v2zhzyUKyy0uukAKoax1cOHqgyE,10915
neojax/utils/__init__.py,sha256=NFOHJ8D1IR1K9c11ELwVLRPR_6Px9cu6fXHCYwEAbPw,327
neojax/utils/torch_converters.py,sha256=V2ysE0hkNzrkABYReICEwrF3rt1m7LodIgteGPblWn0,9402
neojax/utils/tree_utils.py,sha256=QxLrJbDZfpmHmSfFJSwJ0N7IpDaUa0ODwUOVuVKIWfo,2587
neojax_operators-0.4.0.dist-info/METADATA,sha256=Y9wWiduI6k-Xd-R9GQqE7HYcLm4_d76Cj1ycthnG6sc,8078
neojax_operators-0.4.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
neojax_operators-0.4.0.dist-info/licenses/LICENSE,sha256=LVCoe80qIBX8pkbgtL2dNY8HazNsJGGvk67xdpD2pWU,1069
neojax_operators-0.4.0.dist-info/RECORD,,
