Wheelodex » Projects » jax-cfd

jax-cfd

View on PyPI — Reverse Dependencies (0)

0.2.1 jax_cfd-0.2.1-py3-none-any.whl

Wheel Details

Project: jax-cfd
Version: 0.2.1
Filename: jax_cfd-0.2.1-py3-none-any.whl
Download: [link]
Size: 215411
MD5: 72df1d266ae5874fc9ac1d1de4a16d63
SHA256: c0879a66f419561687d92711bd0941f8f774852933cce83313eeec310160870b
Uploaded: 2024-05-13 15:47:15 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: jax-cfd
Version: 0.2.1
Author: Google LLC
Author-Email: noreply[at]google.com
Home-Page: https://github.com/google/jax-cfd
License: Apache 2.0
Requires-Python: >=3
Requires-Dist: jax
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: tree-math
Requires-Dist: matplotlib; extra == "complete"
Requires-Dist: seaborn; extra == "complete"
Requires-Dist: Pillow; extra == "complete"
Requires-Dist: xarray; extra == "complete"
Requires-Dist: dm-haiku; extra == "complete"
Requires-Dist: einops; extra == "complete"
Requires-Dist: gin-config; extra == "complete"
Requires-Dist: absl-py; extra == "complete"
Requires-Dist: pytest; extra == "complete"
Requires-Dist: pytest-xdist; extra == "complete"
Requires-Dist: scikit-image; extra == "complete"
Requires-Dist: matplotlib; extra == "data"
Requires-Dist: seaborn; extra == "data"
Requires-Dist: Pillow; extra == "data"
Requires-Dist: xarray; extra == "data"
Requires-Dist: dm-haiku; extra == "ml"
Requires-Dist: einops; extra == "ml"
Requires-Dist: gin-config; extra == "ml"
Requires-Dist: absl-py; extra == "tests"
Requires-Dist: pytest; extra == "tests"
Requires-Dist: pytest-xdist; extra == "tests"
Requires-Dist: scikit-image; extra == "tests"
Provides-Extra: complete
Provides-Extra: data
Provides-Extra: ml
Provides-Extra: tests
License-File: LICENSE
[No description]

WHEEL

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

RECORD

Path Digest Size
jax_cfd/__init__.py sha256=DSp4S5wjFZ-DDKkBlkrl1WkUrih93sny-pKXzku9REo 687
jax_cfd/base/__init__.py sha256=saS187Tfhj5CTlLHH2yQD8paEmqJMk-QmWW5Bt3LUTQ 1188
jax_cfd/base/advection.py sha256=mHt0cj0ovVabtFDwnGYmA5QU9yBiGc-PqSt9DtEpaN8 16448
jax_cfd/base/advection_test.py sha256=pqTotMQo81rEgkIc-4fe4Vid-Td4bfiT-8QErPNrQPA 16299
jax_cfd/base/array_utils.py sha256=MmyQe0Jjr48rhMXo-t0ZOpNqXrKwP_hm_kIqe1HhKqA 17146
jax_cfd/base/array_utils_test.py sha256=uu0j3Rr0RHFkANQrT9PltnT2JM51OvLlrIHYz4jCMFc 13829
jax_cfd/base/boundaries.py sha256=DSWHldgKu5aRJ2pFWs75zLQwaeA-7WHwngOiZsyufVU 29949
jax_cfd/base/boundaries_test.py sha256=JfoEaWzRuIKlFmPgiRWEcUYWyyTs1jtqu8UfNbHC5yc 50366
jax_cfd/base/diffusion.py sha256=TKprgS7e4P1012sIcrzRYzdnUsJ93GF6ZmVrgQtCoZs 7551
jax_cfd/base/diffusion_test.py sha256=q6T49ucKdOH6rkUjvxObKixNuVZKs79157QBX3VXGB0 4430
jax_cfd/base/equations.py sha256=RqImJWH8spKlH72SPYO54mU0hQuzFqIx0LMjFG1oPKU 6794
jax_cfd/base/equations_test.py sha256=QJ5FQgaTGV-qK5hyO02y5yw-VcRS-7ZLQxmfO65qjKY 8404
jax_cfd/base/fast_diagonalization.py sha256=yNObEljtSU6d6Up0MKGkhlMT8WIWICjxFJXvM1oJvBc 10641
jax_cfd/base/fast_diagonalization_test.py sha256=o0d_b_GcwL2znoz_gmSGVLvsPhmbYz1hg9u9NQpRHkE 4889
jax_cfd/base/filter_utils.py sha256=hOSKeyNW9ZONG5p8YBpd7HT82ggnExr10fr4BykaU7U 1559
jax_cfd/base/finite_differences.py sha256=iXju97txqiGQARhkbVIyAHh7bAHuxtvU765PH2EmBzk 7595
jax_cfd/base/finite_differences_test.py sha256=HtllPRhQ0wEsKoyqZVMih-a43yir7qyAE_SGBSNVMlo 13355
jax_cfd/base/forcings.py sha256=DHUjBbTBvgGsmKasyOJslCrOE3lByuR_Sech-rBNBWo 7467
jax_cfd/base/forcings_test.py sha256=5qp076V5YcoJ2aJ4Ondjx5qegiDfgvAJ4gANviRlQa8 5577
jax_cfd/base/funcutils.py sha256=vEipmJYGAGFcmQmOeQWMM5VcI0kjzTvf3tto3zbd4lA 4144
jax_cfd/base/funcutils_test.py sha256=4Xs1UV1lfATK089EVYkDvmEFWIrz5vvS0s5j9UzZxEM 1801
jax_cfd/base/grids.py sha256=kXgXDmIMG8VPaQgJvoU8Ijq73KXGvX0OPLZz6pLAKlo 24054
jax_cfd/base/grids_test.py sha256=SFBTI3_NRyPAp-v9pNc1GHJhDXh7Jn5mXM3_JEoKLN4 27124
jax_cfd/base/initial_conditions.py sha256=9xtfAuQ7EE68nzNTUEEImw9R_VlyRSUqb_2s9LDDyLw 6213
jax_cfd/base/initial_conditions_test.py sha256=o8ti7CBpSv3NakafwLZz0CwmCYqWjlDoln-8cUgdy4g 5552
jax_cfd/base/interpolation.py sha256=_Z9jjtoDBf3iAOw7eZqUx8_PeMzHXRU7f2YJ0hnCM1Y 13834
jax_cfd/base/interpolation_test.py sha256=pLX8TNiYYlK2OH-B6XYVL77P2aw6JtPOxAP-PpBSpGU 14299
jax_cfd/base/pressure.py sha256=sIKBYNnFatOZ1sJdck3CXcXL8tzTV86enmfwNrB-6xY 6511
jax_cfd/base/pressure_test.py sha256=7zKi89SMwSbEdUdrVZqu1TiydqTJKHtZDrZQnxd2W4I 7936
jax_cfd/base/resize.py sha256=zBjEiTF3y2IcD7LLbVojeXHMeIWQXeQWdSC2yEyO-yY 9351
jax_cfd/base/resize_test.py sha256=TFWqpm_Mn2_6R9BgGbb69EEXaQFt9F0oGonBgyqosoU 15526
jax_cfd/base/subgrid_models.py sha256=eMr1QgdAe3qPCjJab4WI_HolERPUFKjj5J0Z2P5qXrU 9675
jax_cfd/base/subgrid_models_test.py sha256=f1c1lFErSjctJ-mEWHiZs85jSAnlisw5Y2C2kepvnO0 7876
jax_cfd/base/test_util.py sha256=730iPFnIPEgcGTe_bTp23nAHMHPlt_IBC7nNX4WosBw 3183
jax_cfd/base/time_stepping.py sha256=ToNv4_v4GULBYJa1gzheFtywZBreD7nCZw10H2Ze0-Q 4109
jax_cfd/base/time_stepping_test.py sha256=-TaVtH5xCXrNCuyFXzWv-GPsIO7SDNeKA7ieFcwAfGY 2641
jax_cfd/base/validation_problems.py sha256=5bG44Itg0OYQ1rYXzGmGUAGOeGjxFJG0y9xKKcAwnDo 3288
jax_cfd/base/validation_test.py sha256=MpHp79uDp-fU6_Hj-NMdLzoCoLmqoS_0TVUcPJJ-H8g 6860
jax_cfd/collocated/__init__.py sha256=2G4slzSmS775B0FEET6wgTiVyYws4nVUWLZOvWgy5j8 836
jax_cfd/collocated/advection.py sha256=z5xTVT7Ot_AsXuyBaXDK-kLhOnmfb731oYCceUQDDYo 2629
jax_cfd/collocated/advection_test.py sha256=hP4qaPnmr1QdvF1Q6PdnGGRZJv3mXiGD96CtNZ_Y5lw 4403
jax_cfd/collocated/diffusion.py sha256=6RJcnK9GX18g4dd20zIsZ5aYgoDvR4JoOZDQbbLxxmg 1356
jax_cfd/collocated/diffusion_test.py sha256=oyy9vkqHAuwTJV_Lu0e8EMiLZbmTQbRTSCgeL_juiEs 1450
jax_cfd/collocated/equations.py sha256=ax9z48ka-3mmhb-PPpnvBrCI_7vN50Lnf1rgEdWSx30 3037
jax_cfd/collocated/equations_test.py sha256=EZw7hj9tEuKOGHO7xqaknmNWmVi9lA4Gm2AAGh3YIJc 5111
jax_cfd/collocated/initial_conditions.py sha256=cs8j5YvuuVLA2AjJwVg5jhRORf3Ng3kPRdsnx5g49Gc 3780
jax_cfd/collocated/initial_conditions_test.py sha256=3DTM3IAo3b7CH7T_0E5QjNmVx5RgrhEzkHbvrcW-fLg 1988
jax_cfd/collocated/pressure.py sha256=pYJnKJrn8XFX6RntscetEKraqz_mWZv8MMY5jQ5IMOI 3368
jax_cfd/collocated/pressure_test.py sha256=2zJRQ6kKy4bHTlZxKRBQUgqoJLVCMEehtyUzbKjYdx0 2591
jax_cfd/data/__init__.py sha256=BchrKbZN1jE7aTLlf8X1dat9xgb--INWjzKuSLtk9VQ 740
jax_cfd/data/evaluation.py sha256=cUaH4mjLxkBR_RFSfcQzlcpjsZ9pqRvWftglCfcw2Y4 13233
jax_cfd/data/visualization.py sha256=4IY4USqZ48W2ZuEnM-CqoQbEffK9zEsg6OGeWR5WcGQ 4992
jax_cfd/data/visualization_test.py sha256=l5PuesPlNdWtOLbFamcO2Ufn70D3OgZLm99--Hjvvkc 2546
jax_cfd/data/xarray_utils.py sha256=An2Qa4YDAORcuczpKbypa3134kp5S1h23_ofaRsXn1o 15332
jax_cfd/data/xarray_utils_test.py sha256=0y609N0qO5_O-mgj7RqADCfASbcI60y2kbo-VnZ-pJs 2780
jax_cfd/ml/__init__.py sha256=-Oa5DkiaXAKtU-j9tlU1rxfr8FppQfi7Bdba97UpTP8 1246
jax_cfd/ml/advections.py sha256=KJdupucel7icZyuOPSj33YSZHf0kl8WhuLztBDi3DfI 2798
jax_cfd/ml/decoders.py sha256=1pnwYGKtCPBL7byLLYxyDUNawNnYC4mgKqu7DtlQ8yM 5480
jax_cfd/ml/diffusions.py sha256=397TA_l9t_WiJ-bumf2Dr9wQcccF_at1Pmqym3STnMU 2874
jax_cfd/ml/encoders.py sha256=tfv18ZvaVPMPiwRIcBYEvQtPtAJJy5eMBNoblmsOLk0 8939
jax_cfd/ml/equations.py sha256=ULhg8Y5RsivAGVDpRyPrifrK43HFIma0UlbwJGuFLeg 9351
jax_cfd/ml/equations_test.py sha256=T9si2nj0njN29Hj0U4PqERJ_JxAWiqu2XfEGRw0tnBM 11960
jax_cfd/ml/forcings.py sha256=Ha9Z8L2mUx83KCEcmFJYIuLN0c9ExTuZh93paoFWrlc 3550
jax_cfd/ml/interpolations.py sha256=XBwQ37xn29znL6FVrVquWPQc-aA0sJCAVel32ewvB6o 8597
jax_cfd/ml/layers.py sha256=YwrXU8QLp21wvQZJO7XBttl0ezp4KTGd9LL-KrW0Uto 27978
jax_cfd/ml/layers_test.py sha256=SB5QijiRMkdZRTWpHPC4XLniTARCeqz0Mv4nMi4AzHU 20039
jax_cfd/ml/layers_util.py sha256=ag7pivKglgZYmFNon_rSPAVMtrZSquu_lIeAIuhSIN0 15853
jax_cfd/ml/layers_util_test.py sha256=z1IZG-eu-e0F_ynt16gW08tJXSDbxcJUZoeffh5HTYs 12918
jax_cfd/ml/metrics.py sha256=dJuctYsTf-YcKtdF9q3cnVNAoma2zSCfViMIimbCSMs 5667
jax_cfd/ml/model_builder.py sha256=SAuMyQFxXi1sfkLH5pVeNKa3n_76WIpExJowrtfSwgs 4829
jax_cfd/ml/model_utils.py sha256=JllEVnYCT8pmP98o1CFaJQCevpdubHo_vd3jc7J9JtM 3033
jax_cfd/ml/networks.py sha256=Nwess1xBgqEgeAsQiP6qWxkCBWtoesafYFDSZO3j1_k 8726
jax_cfd/ml/nonlinearities.py sha256=gIFZYxNarH6bfY0m-pwGd9TpxvbrQWUARIdEMq0Ovrc 400
jax_cfd/ml/optimizer_modules.py sha256=369JnHbz9uqVYsNw-vLonyWWk_YeA8u00TXH8tG9HWY 518
jax_cfd/ml/physics_specifications.py sha256=gxzlhSuzcygQMKBNp9vhnV9t1H1TfKCCF3C472txo0I 1301
jax_cfd/ml/pressures.py sha256=BQHuMTGVqPSx-QZiyGRDDxYTCKBGG6OPRV-nUzqCyS8 905
jax_cfd/ml/tiling.py sha256=1Bp5eIPIbbD430HNaKlvnxI4tJBWYITr_lsBpVONx4U 5464
jax_cfd/ml/time_integrators.py sha256=xUvpduAebzPhbyfp6qGWU4y5uez2n1R-Ursr9Uq26n0 967
jax_cfd/ml/towers.py sha256=n_ZNiPOiFURizfTUEydNG_Rw8Li_VM1tn5bmHMLJQzE 11396
jax_cfd/ml/towers_test.py sha256=eXBjpJsPmFd088tGHhEGF3hSNApxGxWolq6eLAr2z64 2736
jax_cfd/ml/train_utils.py sha256=qJqUAD4wE0SrY94kGDi-Vnb9Pf41z5zNOHfhY2PKrP4 8036
jax_cfd/ml/viscosities.py sha256=6Vs6_Fy6V3V1LMTkCAquUZWIM5w5N6jcm7s9N6SpmUI 9873
jax_cfd/spectral/__init__.py sha256=dMpyq-Fwek7lmtEuKS65wtEC4MDbjZcdNtL0eTnX5ws 745
jax_cfd/spectral/equations.py sha256=9B818IcqGDzcXBG0ssfGb5E2bEMtoKOb8OOuWmpddfE 9013
jax_cfd/spectral/equations_test.py sha256=yzcbGXaSjAVlHw0XntEr2xFFWV0ljzImVM3ej3P-gDs 13756
jax_cfd/spectral/forcings.py sha256=HwbXJPwTQVDRWk4WYAh1-rRoJVdgJIeGN4mUNbReyqw 1572
jax_cfd/spectral/time_stepping.py sha256=Ee3myp8MIqGgs1E988DW75akglj20GEm1H6Aytj87gk 8747
jax_cfd/spectral/time_stepping_test.py sha256=VqPFjj0kGOJNyYzcqMu9QGccxMdUZZLQTQCWa8cfkvk 7062
jax_cfd/spectral/types.py sha256=2loQ_aE_mHufU1C7JhQgQJHwpR0gHozkBTwCYIrKuXo 779
jax_cfd/spectral/utils.py sha256=qkoIyH02h4wFoxKgFS8KcKQDIcHdh7J4m8TQP0IHj-c 6804
jax_cfd/spectral/utils_test.py sha256=NwHWpA4Yn91_3cdPHQfS-sKT-BXBuSGKY-MaEGOftJw 3063
jax_cfd-0.2.1.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax_cfd-0.2.1.dist-info/METADATA sha256=6kzMOf290nWXAh8oB612r5WDMfoz_ik1Jl8wTvqEzOw 1355
jax_cfd-0.2.1.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
jax_cfd-0.2.1.dist-info/top_level.txt sha256=NLKZZZVwSF6wSaPOqYwIcc8m2mS-v7gvYfHLCOO03Ow 8
jax_cfd-0.2.1.dist-info/RECORD — —

top_level.txt

jax_cfd