neojax/__init__.py,sha256=dHYgOnzsXfO54v6t9ibywCNzqAlCoV0PH0acj4vDojc,83
neojax/benchmark/__init__.py,sha256=uUybbmus0e3XtmlhEVWRqVSubFY7nDVXGC5AYx7bZ7I,531
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/array_types.py,sha256=OBZd5qyxSaL_ngolXV-QSFC32aT-shlAT877uHRaLps,802
neojax/data/processor.py,sha256=24MrnQWhGHg4x7M7KICcX5OrrbUDGITGTuQ7y6G_gYo,5823
neojax/data/scales.py,sha256=JOUO2LbajBZ0LXZbBFXwW557R9PciKGP9h2eXvN2XUk,4884
neojax/data/bundles/__init__.py,sha256=89afchgmjKIq7PjFZ3XlTsC5oWYz6lcvq6hNCpLeCYM,116
neojax/data/bundles/data_bundle.py,sha256=SGmImkV2twA0NG6jLssKuf11vvKycCBz-E7PkscYAkw,3463
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=OwomHb1xQkq1vZ38-kyIaPlAEp7Mgt4iTaIPT1wWlcY,12576
neojax/data/datasets/raw_dataset.py,sha256=Nc050a_YsAA5BQfKXLCsb5JED8NH-maixm3zedFcNM8,7269
neojax/data/datasets/utils.py,sha256=TnCST9uqfJJeo9CydU5jrbHhaf4tr-l2zZQXzFYAi2s,19889
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=L0qlH5xV12ADNMxOzJP5RtvLfT-RqrBzmz1IEJTKpjE,11896
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=JakYtcRE953GjbLUsqjbVPzKAxzCc6Qa-I1peadYVVM,1184
neojax/data/schemas/bundle_reconstruct_schema.py,sha256=vIcFgab-RBspEo_mfHr8LrBKl5jcYjE9GmKsw64-F6k,3274
neojax/data/schemas/composed_schema.py,sha256=odReNp3s2f8UeXGxZcLLPFE5aFu6CmR-9Evphh08mcA,3639
neojax/data/schemas/concatenate_coords_schema.py,sha256=9sen_hiMqUzO3AhACuim_XngW5KKAiGmo_0WyeNd9as,3909
neojax/data/schemas/concatenate_params_schema.py,sha256=B_iIBkouTCccYPUSZVReYD5s4DGYle9ruYcQPQ-qOOw,4083
neojax/data/schemas/flatten_time_schema.py,sha256=pg_RepVnMVVhX3Zzler_h-gZ94M2__EgzASyHobXfUA,2375
neojax/data/schemas/flatten_to_points_schema.py,sha256=9qeB2_q4yFybVRFuIRqx1yoPp9rDqWkQyM8oxGR7Km4,3111
neojax/data/schemas/graph_tuple_schema.py,sha256=9lHDwNA5SBd0358kexWzPPioUXOjFiyj5-iqpfhpsUI,3969
neojax/data/schemas/identity_schema.py,sha256=55VTixUIfhYt2V4EIomlASBqoMXqBbWi_8lF5cCiGSs,882
neojax/data/schemas/mesh_schema.py,sha256=ScsxgW5eNdB9gFifkdOwDRnEU6ahkH2GI35rQ3vndps,3864
neojax/data/schemas/reshape_points_to_grid_schema.py,sha256=1fedgduKTseWMBgco-E86TpsS8Vcm5axjowBjcUx5-s,1979
neojax/data/schemas/time_to_stationary_schema.py,sha256=DmGajuVR1_NyY63FylYQ_OfNV6jgQE19WjxtNp9gBnM,5274
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=lfgO0jOHnq92ocU9s6qz7fzgr9zbntrCy20V1CY2ThU,493
neojax/models/baseno.py,sha256=ScWVymKOKOFOAFo2F7FadTWBDct5LtZUNsOekQK_7V8,10985
neojax/models/deeponet.py,sha256=hpqjI0yA_T3cWyGUg389r8QSfh0AP4wdQb8NXviP7a0,6616
neojax/models/fno.py,sha256=-7SF4XYpsYi4klP79OZ45bdUDKNo-oxfTAYtlHRKeas,17554
neojax/models/geo_fno.py,sha256=pizQaGDrCMIHyDE06o2LAIvYP2VT5Gl84FrNxKPyggM,15759
neojax/models/tfno.py,sha256=SRiKJejjc_v6FnQYMOfQPtK7bkPYZ7pxrTLlT8mIDo0,11744
neojax/models/uno.py,sha256=7cLYHvgaFPwOIm9hfQnaKgWmuk3Um06g0sM3GkAqMJw,21244
neojax/nn/__init__.py,sha256=XQsZa1KeDVAp2CT675DFc7zT2FxlYyO4kydv9hdxZSg,1204
neojax/nn/domain_padding.py,sha256=swnKUC_hUXUeJ-jrLEskyzjCUvQUBSrD7yQfciPZSuA,7410
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=CBTFmXgdvDoyH_kV8tIQyMfqF0imJRh5bN-Uq364xDc,6929
neojax/nn/normalization.py,sha256=RhK_EinIezFctt4i1cmAHq_-Ic-ALjZWIw8x_A6GAIE,3052
neojax/nn/pointwise_mlp.py,sha256=SO2ramfmTV1h9ZbXVHeZ_zFnxZBKleFVkdcG3sWSAuU,4989
neojax/nn/positional_embedding.py,sha256=znEGFNFCyML5EuKZUjqzKR4OegJI7q22I0m8PiBy9rI,2732
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/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=O53000NsOq8X9ET7Hlesd601maAwqNFDagvGANXf4Eo,10940
neojax/utils/__init__.py,sha256=nhztg8QdzMwED1yCPqiee6PBj-Efi204b-06sEfPpzA,240
neojax/utils/torch_converters.py,sha256=Z2G85ShiGft3cRFnNOg9yhvcm9u1QZ0LlItlXhxHbi8,8894
neojax_operators-0.2.1.dist-info/METADATA,sha256=m2k_p4Y8_TJCqDc8bF6rhKeevxBp6n9oZfdwCUtforg,6183
neojax_operators-0.2.1.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
neojax_operators-0.2.1.dist-info/licenses/LICENSE,sha256=LVCoe80qIBX8pkbgtL2dNY8HazNsJGGvk67xdpD2pWU,1069
neojax_operators-0.2.1.dist-info/RECORD,,
