jax-cfd

View on PyPIReverse Dependencies (0)

0.2.0 jax_cfd-0.2.0-py3-none-any.whl

Wheel Details

Project: jax-cfd
Version: 0.2.0
Filename: jax_cfd-0.2.0-py3-none-any.whl
Download: [link]
Size: 197002
MD5: f0096ea542983362e1b5264e482ef5ae
SHA256: 90908c64937942075736d56e5c325bc58f251da8f718bfb6d623ed784a68c069
Uploaded: 2022-07-19 17:29:30 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax-cfd
Version: 0.2.0
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.37.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jax_cfd/__init__.py sha256=gKsu0J61IiKjh89kCpG7LzPH7eFweGjpDEQFEFfX4As 687
jax_cfd/base/__init__.py sha256=saS187Tfhj5CTlLHH2yQD8paEmqJMk-QmWW5Bt3LUTQ 1188
jax_cfd/base/advection.py sha256=HFKQFKceSXTa9sb-Uu4tD0OPVp2ip5T2I8SsFHF0V24 13534
jax_cfd/base/advection_test.py sha256=vQCV2fUaAL3ZQU_a3nugHoKC6qdxawChP-SzDA5QnQY 11537
jax_cfd/base/array_utils.py sha256=3PS9ecIkYOZALfLqDXju1-PGhoLJeXmZUBWTvQniW7E 8119
jax_cfd/base/array_utils_test.py sha256=4qp_-lYxkn4xmd578NVg1pt3QvI1ADvR89l966HMLVA 5902
jax_cfd/base/boundaries.py sha256=R_biA_ia2n1IaWzPSWUOLg8ZHVmx00SKEEL4d62tjgU 12379
jax_cfd/base/boundaries_test.py sha256=66bp6MsKrk_s--wLJ6f-pE7pDNrG8kRUn_6EECvSUvc 26736
jax_cfd/base/diffusion.py sha256=zpAOzHTy7cHE33rb8IurTcZb02aan0KwbZtOh8KmJhs 4157
jax_cfd/base/diffusion_test.py sha256=NNfXFXfq2LlxDfHswtwaOQR-YxZj7Tjf_5Q9TaLmFCI 2215
jax_cfd/base/equations.py sha256=i55T2PCKzZtukKxh6NosWtnjS4n-qZhADF-FpTBDzgE 6746
jax_cfd/base/equations_test.py sha256=QJ5FQgaTGV-qK5hyO02y5yw-VcRS-7ZLQxmfO65qjKY 8404
jax_cfd/base/fast_diagonalization.py sha256=iYWzxoxwutPPFND-n-P9ZKlc5JbJbrKqVOekXx4wkKU 10647
jax_cfd/base/fast_diagonalization_test.py sha256=tCvLZIgUTqhpW6KR1Hatfq80rml6aaj2JQdIYT7eI0A 4889
jax_cfd/base/filter_utils.py sha256=hOSKeyNW9ZONG5p8YBpd7HT82ggnExr10fr4BykaU7U 1559
jax_cfd/base/finite_differences.py sha256=NzHQ4B6kx-J-Fh9wqVk4tT3RI41abf9Ivn-Cs1rAWvE 7323
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=R_nzR--Xja12Bs3Ny0JQ9Pf3hH5zpYpNXef80Ylol0A 22869
jax_cfd/base/grids_test.py sha256=ue4bu6CE4bxGIP-42BKUm9dA-sZnb8p5lVcp8AaM3vY 27120
jax_cfd/base/initial_conditions.py sha256=jFQEr-xg8W6LWJNJt6bPsLJ6NylhuUWVRbsbDFuJnZQ 5938
jax_cfd/base/initial_conditions_test.py sha256=yjMviO5HTT3Q7ce_-Mtrbu00VRzEVoIfCVXOwbJIFTA 5541
jax_cfd/base/interpolation.py sha256=S7JjiVr63NPXXHSfw7uMzsEYPCuRJS7SoUZvrSCu4co 13846
jax_cfd/base/interpolation_test.py sha256=pLX8TNiYYlK2OH-B6XYVL77P2aw6JtPOxAP-PpBSpGU 14299
jax_cfd/base/pressure.py sha256=ZJj6hos_erlHkUPVV1tj_h_nRoA0K6QOpXxtv0K_R2E 5008
jax_cfd/base/pressure_test.py sha256=0elcoaYYNXfNsldVKaaJjHVqMuJUQqZFfgRekPQwIMU 6300
jax_cfd/base/resize.py sha256=2cxvnUYER23TO33KFlTs2x1swWlgtmgc2KMMiCdJI6w 4764
jax_cfd/base/resize_test.py sha256=H9AJX_2sBB4ggWe6VyMDOZr64LmFUFFyFZ_JqYAupdw 6872
jax_cfd/base/subgrid_models.py sha256=hM8p6nQAuC_dfNpVKYyZs5BqZUymFk7oMPC64WdElcY 9683
jax_cfd/base/subgrid_models_test.py sha256=f1c1lFErSjctJ-mEWHiZs85jSAnlisw5Y2C2kepvnO0 7876
jax_cfd/base/test_util.py sha256=qobOVcGQMG2G3_PTmBBuz_dvRN37P-lTB-XHrLJpwO8 3191
jax_cfd/base/time_stepping.py sha256=ToNv4_v4GULBYJa1gzheFtywZBreD7nCZw10H2Ze0-Q 4109
jax_cfd/base/time_stepping_test.py sha256=rY7wid04d7-OcQB0-Qny3-fa1-U9C3UXE2sPAWJREYw 2648
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=GL9sEG6WPRL_URwRruC9Qe5ifPXuzSXOGkyvXxfLvlw 2427
jax_cfd/collocated/advection_test.py sha256=g0z0sf4rVjvJ8o5IDg8oDMVnyR_xBcWrlXcE5plQvZ0 2245
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=2C1Purj8axHTwyQIxNu1TsELPIFmJ9VkID_t3u_xiaQ 5111
jax_cfd/collocated/initial_conditions.py sha256=1NkxBh0KOMNuJVOji9RIlESmJPB1-RAXqRMbI2eRLf0 3775
jax_cfd/collocated/initial_conditions_test.py sha256=zypgf8traB7H3nVlmEYwuAyoB-Mej4H-8Ui0FI53W6c 1977
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=EIXvYeUL9wM6JL-CoQuUjlEEye6gVElF8qnWtNwtdS8 4950
jax_cfd/data/visualization_test.py sha256=l5PuesPlNdWtOLbFamcO2Ufn70D3OgZLm99--Hjvvkc 2546
jax_cfd/data/xarray_utils.py sha256=W7QPft9r3oQdBkXT4QB0X9Y9Jh6GZr_7Lr0BTuEuUCk 14868
jax_cfd/data/xarray_utils_test.py sha256=JbzlC8GYTKXl0anOk7MMi8nFMWGem9ma9FQsgHaGFJY 2721
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=CvaafcirlYufej7HqvQCVX9ObhzMYKN8lgPKYDF22O4 8753
jax_cfd/ml/equations.py sha256=ULhg8Y5RsivAGVDpRyPrifrK43HFIma0UlbwJGuFLeg 9351
jax_cfd/ml/equations_test.py sha256=UelA_U2VkM2fvYTjQ-agHox_UWtJU9K_qXEQaJdDPJg 11830
jax_cfd/ml/forcings.py sha256=Ha9Z8L2mUx83KCEcmFJYIuLN0c9ExTuZh93paoFWrlc 3550
jax_cfd/ml/interpolations.py sha256=XBwQ37xn29znL6FVrVquWPQc-aA0sJCAVel32ewvB6o 8597
jax_cfd/ml/layers.py sha256=IroFxhpp8R3zZ_Ed_1bi5JEl-A-HC1fYTW01iMdoX1Y 24951
jax_cfd/ml/layers_test.py sha256=SB5QijiRMkdZRTWpHPC4XLniTARCeqz0Mv4nMi4AzHU 20039
jax_cfd/ml/layers_util.py sha256=C_dPBo6DZqBwKxpflJORM5uTn6BKdz3JDHYZhi0lGZk 15790
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=qj_okxL1VuEBNMEjWYe5Po9umYIV49fSqhxT9SrDJVc 6592
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=cQYUkdLulUZAVXW_aAcoz9_8tsZKN-v6Gzf2L6pRVzs 10752
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=TMUwEXbi1tnuDyTC4Anh3sn0ZjAB5Jhj6MZKQ0SA-B0 9708
jax_cfd/spectral/__init__.py sha256=dMpyq-Fwek7lmtEuKS65wtEC4MDbjZcdNtL0eTnX5ws 745
jax_cfd/spectral/equations.py sha256=kGnX-RZ9fSX0U3moUmKQcsB0TNmvjCB5MGxT09H9GBI 7696
jax_cfd/spectral/equations_test.py sha256=6Gv18c8GRkLbArIGG5l-EBbVmhD9ESZhcCCgEmJabys 11933
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=pvwjBC76OXD552aBVUZL5ay41mhMZfnLika_RyfGhfw 7069
jax_cfd/spectral/types.py sha256=y6NDhCA_YgpmDt5At0SqCJ92TIINrQAdsVB5KraKsvw 798
jax_cfd/spectral/utils.py sha256=ZGf3GS_OBuZax7kDuwTQJNphmeO_KuzVVsd2cl-TC48 5631
jax_cfd/spectral/utils_test.py sha256=NwHWpA4Yn91_3cdPHQfS-sKT-BXBuSGKY-MaEGOftJw 3063
jax_cfd-0.2.0.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax_cfd-0.2.0.dist-info/METADATA sha256=pH_IVgNJeCCeOhrL4zLnQcnd4mrgXOTMwuMjkHTrQ_4 1399
jax_cfd-0.2.0.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
jax_cfd-0.2.0.dist-info/top_level.txt sha256=NLKZZZVwSF6wSaPOqYwIcc8m2mS-v7gvYfHLCOO03Ow 8
jax_cfd-0.2.0.dist-info/RECORD

top_level.txt

jax_cfd