jax-dips

View on PyPIReverse Dependencies (0)

0.2.1 jax_dips-0.2.1-py3-none-any.whl

Wheel Details

Project: jax-dips
Version: 0.2.1
Filename: jax_dips-0.2.1-py3-none-any.whl
Download: [link]
Size: 129540
MD5: e5c3a05b30f69db8dce58e2d8ce078fd
SHA256: 1216119750ddf90311277c1dfdcc10cd9e8398673f28efa1bc65fc2fa9110690
Uploaded: 2023-07-06 02:24:48 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax-dips
Version: 0.2.1
Summary: Differentiable 3D interfacial PDE solvers written in JAX using the Neural Bootstrapping Method.
Author: Pouria Mistani & Samira Pakravan
Author-Email: p.a.mistani[at]gmail.com
Maintainer: Pouria Mistani & Samira Pakravan
Maintainer-Email: p.a.mistani[at]gmail.com
Home-Page: https://github.com/JAX-DIPS/JAX-DIPS
License: GNU LESSER GENERAL PUBLIC LICENSE v2.1
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Financial and Insurance Industry
Classifier: Intended Audience :: Information Technology
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: GNU Lesser General Public License v2 (LGPLv2)
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Physics
Requires-Python: ~=3.8
Requires-Dist: jax
Requires-Dist: dm-haiku (==0.0.9)
Requires-Dist: flax (==0.6.10)
Requires-Dist: optax (==0.1.5)
Requires-Dist: jaxopt
Requires-Dist: cython (==0.29.20)
Requires-Dist: pillow
Requires-Dist: pandas
Requires-Dist: pytest
Requires-Dist: matplotlib
Requires-Dist: meshio[all]
Requires-Dist: gdown
Requires-Dist: warp-lang (==0.9.0)
Requires-Dist: pyexr
Requires-Dist: trimesh (>=3.0)
Requires-Dist: scipy
Requires-Dist: scikit-image
Requires-Dist: six (==1.12.0)
Requires-Dist: moviepy
Requires-Dist: opencv-python
Requires-Dist: plyfile
Requires-Dist: polyscope
Requires-Dist: mesh2sdf
Requires-Dist: hydra-core (==1.3.2)
Requires-Dist: hydra-colorlog
Requires-Dist: black (==23.3.0)
Requires-Dist: flake8 (==6.0.0)
Requires-Dist: isort (==5.12.0)
Requires-Dist: pre-commit (==3.3.2)
Requires-Dist: biopython (==1.79)
Requires-Dist: biopandas (==0.4.1)
Requires-Dist: rdkit-pypi (==2022.3.5)
Requires-Dist: pyevtk
Requires-Dist: jaxlib; extra == "cuda12_local"
Description-Content-Type: text/markdown
License-File: LICENSE.md
[Description omitted; length: 8211 characters]

WHEEL

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

RECORD

Path Digest Size
examples/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
examples/stars/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
examples/stars/_solve_single_star.py sha256=G-LZH0twTdqbQWjEFOFlxz9qO7TUIPc01O6Cz9HZX3o 7157
examples/stars/_starbox.py sha256=I9hlgLj4Sl396LR6Tv85cGPmRJ-Q9n9n5yAshgB8qU0 8780
examples/stars/solve_stars.py sha256=_LKg9Uh0aUTtDMcdLbR4RESmzp1t814U5_aoO9OaHws 10881
jax_dips/__init__.py sha256=sXLh7g3KC4QCFxcZGBTpG2scR7hmmBsMjq6LqRptkRg 22
jax_dips/_experimental/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/_experimental/compositions.py sha256=tDBKDLupQy-dLzzhNuZyCSU6eXzIt2U_AsT96kTLdiw 3244
jax_dips/_jaxmd_modules/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/_jaxmd_modules/dataclasses.py sha256=H5-kescZX_FAIKBHu8dcXmrGkEENRwb6UxUvoRZnhAU 2379
jax_dips/_jaxmd_modules/energy.py sha256=7dFyUd8hh-p7VC3KAABUYWbKHonhV6fYPCKW1bRaDhE 4620
jax_dips/_jaxmd_modules/partition.py sha256=4yBO_tTg6f6ENLu41OqDDwSP5BfIdwxbiLDAD1Su0bk 23628
jax_dips/_jaxmd_modules/quantity.py sha256=QvoOAo1NcBJNhppJ9jAM8getmkwsNd-VCisoMbgjnys 2571
jax_dips/_jaxmd_modules/smap.py sha256=Y60p-pLeXBgFBA_6aKBbL37DUPlIL0zw88TJaRQxDxQ 7086
jax_dips/_jaxmd_modules/space.py sha256=6McV4wURIhEDWa9CE16fsfChkXfu2wmV-Hwt9ht6O8I 13574
jax_dips/_jaxmd_modules/util.py sha256=JNFBlVANZVcQNkq7TnK3XrIcJJ_K_pDGxWHHYZMe-HM 1863
jax_dips/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/data/data_management.py sha256=-McQeODPu42aITZbhObcDY0hYj7dCZ9fCXpnEQgxd4M 13349
jax_dips/data/data_stream.py sha256=jD1EVshRGfSB0SRuONZpgd55b_uqXMwQFW-Gk7eMWVU 3452
jax_dips/domain/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/domain/interpolate.py sha256=s7NYsc0xGf_KTsspQlTTgpwoBY7m-fKJyVBozqw4gT0 40392
jax_dips/domain/mesh.py sha256=Pm11fntUFWPpo73eKV-zq60v8OrTl2ebA0nS-kq1D_E 5009
jax_dips/geometry/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/geometry/geometric_integrations.py sha256=67Le1YBWNaWnq0_YNVwhVPMBGA63Fjj6au5E43lLvyM 39302
jax_dips/geometry/geometric_integrations_per_point.py sha256=J9XvY5G2laR-54EOZcGY1wp0IEDR1GqnqhjQTdjaooc 37950
jax_dips/geometry/level_set.py sha256=okqjBiZCjz674CapmVmj8FbzTdjcStzB6ERAWvBhKKM 9824
jax_dips/nn/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/nn/nn_solution_model.py sha256=dmgXXqU1Thy7Q17yiOOe3cIqWzgNASfZuJq2Y52sMDk 4493
jax_dips/nn/nn_solution_trainer.py sha256=ThacL7b5kI5Xglhk6NW5LmuPiyIrKsXAhCvdMe6heoI 6619
jax_dips/solvers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/solvers/optimizers.py sha256=EWSRPSkmQswA-nI4WtDWzbUndaoymsVjP-90iqNzhq4 2479
jax_dips/solvers/simulation_states.py sha256=iU8AiMCnOFxDijluUSBUzKaH4t6SRYGSFxwxOm5EhNI 2544
jax_dips/solvers/advection/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/solvers/advection/solver_advection.py sha256=aPWkaBtOgp6RD5ZX30eL5RchLIDhjFBPXpabUy_WJ6w 11932
jax_dips/solvers/free_boundary/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/solvers/free_boundary/solver_poisson_advection.py sha256=PbeNWb9_iARR6Hln-jQh3BgquCD3d71NaowMdEgWj6g 29515
jax_dips/solvers/free_boundary/trainer_poisson_advection.py sha256=k4cVWQBKBlW8nHEJpiRLCPRv73YYKp4GAdenqg_57K4 23584
jax_dips/solvers/poisson/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/solvers/poisson/discretization.py sha256=Rl9c3Ft4CdzcChzzvssO0XyqtEFAc2soKhuVsyQTHro 23239
jax_dips/solvers/poisson/nbm.py sha256=nktv-yFWaO-tnNeZoMs4_sfg4LMBOckbXSW_0fReRwg 9023
jax_dips/solvers/poisson/trainer.py sha256=24ZIjkabDBzMXRKHfedCP88CVHjs7b2Wxsj4XuRXZv4 24448
jax_dips/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax_dips/utils/chunking.py sha256=lPzyutFrK-vOkV-ddjDpUlTxb-YMVVSvydpld5hFLkA 2663
jax_dips/utils/conversions.py sha256=g0jcMlvPz2MfGvc-xgehMO15XawQVEw2SrUfdy2hQHk 2293
jax_dips/utils/inspect.py sha256=MbfDxacnOBI1trFqlw-vTHxIbz17JoYgTdgDZ9jYglI 2309
jax_dips/utils/io.py sha256=bA9ZhgQwTIq_HzLDcuNyEROUxqrrpwz_Jo9P5_gr2ms 3870
jax_dips/utils/visualization.py sha256=TKs26gb2ujTWa_D04Zy9WvZuktXRwolHJdAwBmaWws0 6960
tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_advection.py sha256=l2eD6UVn-dgcmFfGzD_ejgTibBCoxzmB30OjphcrglE 6430
tests/test_amr.py sha256=AO7auuWuWE1_zqJvcADH2yQNdC-3YlrNyrbLmhR2pB0 1771
tests/test_geometric_integrations.py sha256=Jph1nj-2qNfnIOmr7f4y6rjZp1kLczjkgfHtmLO0hUg 8608
tests/test_poisson.py sha256=o_BfsBitjRSXiIOTTcXyiKbBkVQlsUCN2RusCYoKj-s 9955
tests/test_reinitialization.py sha256=61y-Yn0V-TBMeqvI2JRojEMpS97XoOGzGE_n_udAoQo 6397
tests/confs/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/confs/experiment_configs.py sha256=3DttUqWw3pk_ifQHZWQKZlm1zbG9suhWZyr7fTwfn_E 12910
jax_dips-0.2.1.dist-info/LICENSE.md sha256=xdpE9QI-KVvtAb4KqxcSb_YQ7Vv-3kxMz_lm7B60cG4 26588
jax_dips-0.2.1.dist-info/METADATA sha256=k7BCThJeEhPrOT0km-h01-dfno6f3-cNMvZPhSi3o34 10373
jax_dips-0.2.1.dist-info/WHEEL sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18 92
jax_dips-0.2.1.dist-info/top_level.txt sha256=UpL_k8tLxWhVViGR8I4HPf0p-HbAZDCSgLdaElKU__o 24
jax_dips-0.2.1.dist-info/RECORD

top_level.txt

examples
jax_dips
tests