Wheelodex » Projects » reax

reax

View on PyPI — Reverse Dependencies (0)

0.2.0 reax-0.2.0-py3-none-any.whl

Wheel Details

Project: reax
Version: 0.2.0
Filename: reax-0.2.0-py3-none-any.whl
Download: [link]
Size: 36694
MD5: 642447342c68ee65886347ce4fc84645
SHA256: 019c72b3d40b8c37961447ae6b28b5caa306caade5b2666b0365fcc4f501372b
Uploaded: 2024-09-04 12:29:11 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.1
Name: reax
Version: 0.2.0
Summary: REAX: A simple training framework for JAX-based projects
Author-Email: Martin Uhrin <martin.uhrin.10[at]ucl.ac.uk>
Project-Url: Home, https://github.com/muhrin/reax
Project-Url: Source, https://github.com/muhrin/reax
Keywords: machine learning,jax,research
Classifier: Development Status :: 4 - Beta
Classifier: License :: OSI Approved :: GNU Lesser General Public License v3 or later (LGPLv3+)
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Requires-Dist: beartype
Requires-Dist: clu
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: jaxtyping
Requires-Dist: optax
Requires-Dist: orbax-checkpoint
Requires-Dist: pytray
Requires-Dist: equinox
Requires-Dist: tensorboardX
Requires-Dist: tqdm
Requires-Dist: flit; extra == "dev"
Requires-Dist: ipython; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: pytest-env; extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: pylint; extra == "dev"
Requires-Dist: black; extra == "dev"
Requires-Dist: docutils; extra == "docs"
Requires-Dist: jupyter; extra == "docs"
Requires-Dist: nbsphinx; extra == "docs"
Requires-Dist: pandoc; extra == "docs"
Requires-Dist: sphinx; extra == "docs"
Requires-Dist: sphinx-autobuild; extra == "docs"
Requires-Dist: lightning; extra == "examples"
Requires-Dist: pillow; extra == "examples"
Requires-Dist: torch; extra == "examples"
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: examples
Description-Content-Type: text/x-rst
[Description omitted; length: 70 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.9.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
reax/__init__.py sha256=dJQ9BwylioaiNutZTurUFda8tTqA46afQxM_hzCYWXQ 918
reax/exceptions.py sha256=l2d0Z0wu91TfJl_smSKQXME-9Nglxajn4xbCE5nIo-4 123
reax/hooks.py sha256=mltvZNuhhd1u9YciEYy5CCcKoxfHKUkl7zr9PASeP7Y 1541
reax/modules.py sha256=l3qJ8tkCzAaIWT0tMTeiPm8jARsY4vkUK_AsPH_PETA 4576
reax/optimizers.py sha256=p5-tZaSK7-yaIOdERaJpXljr8bY6DYbOfGeiFfi6KZ0 866
reax/rank_zero.py sha256=cJ-1GwDbMtPAWb3DXboVw60jGWI0iiAU3MWaecxCRzA 411
reax/results.py sha256=aisve7XAQgE3oqKIHScXO8IFhl8P4XA_KlCOrfkZ-7Y 3060
reax/stages.py sha256=HiS3XeJYKfT5FTrS4ZzugHlPh7JT-p7KBwbp-asAT7Q 14535
reax/training.py sha256=YMPQAd6NEOje4WMW4_lsirWXyZKfexShtVWGTVB8JaE 6641
reax/data/__init__.py sha256=Zgq07l8NWgA0oVtlYn3OK51pdcSRjUfCoL4eHRvi800 328
reax/data/_loaders.py sha256=Sd4YIMK8Xuy7w_a3MJelI-qVbOgt37q_aQXVbT9TTYs 3974
reax/data/_types.py sha256=uwDRCbTxFc7MkVYsNeDbmAibmMg_y9uDcTzWkit-icU 361
reax/data/collate.py sha256=mq3p8vImfwPf7JLj5N9WbZflmI5as5UlD-HwlQv9Ylo 5174
reax/data/data_loaders.py sha256=Hy7x0LemlQL49HTUvx8C4fyGlG-VbxQhGE03cr6Gmm0 893
reax/data/data_modules.py sha256=KVElVKF7rZ_Wri40oAap5aYquA52D5jA0dpwJkuuHvI 2483
reax/data/fetchers.py sha256=VkHN1dNKGURW0AGU7_nQAc1yIBdGybrUd5FoKPGm2uA 2748
reax/data/samplers.py sha256=2gmJtdGjjUm7QcDS8N6SdVy_sdDsJ3ZHMVBMiK4k5RI 6265
reax/data/utils.py sha256=WrkJ2eKDsiXco7-b2Alj20wim9yZr-EXLteRx6VWbFk 3274
reax/listeners/__init__.py sha256=4zx2MsvMffJ_O74Hi5yfBulLnKVVaZWxo4CdDm3WuOM 219
reax/listeners/early_stopping.py sha256=BaD5EfwphUSyAbpxTYA27hTMsF4WPMAQg7VlAlGIuO8 1711
reax/listeners/progress.py sha256=xg0ufViaUZW5M20ln9mCMPFwXJxMnxYo9g-8OoBJepU 1739
reax/metrics/__init__.py sha256=_FL0kqFscn-6Fu2XoRU6bD8Ig-rhK6aQrplUctehFsM 443
reax/metrics/_registry.py sha256=l4ShTIIBuRaNZMlmYXi3cqK_cSbfEPx81jeZnxYYFJo 1170
reax/metrics/aggregation.py sha256=V3FdoeUkTDMLqbHAvp6_iHj3qmslitjjrZozFGA_YKo 4814
reax/metrics/collections.py sha256=yKP-LDm3DumsHC9FEIU4h9dSPHvaebtBXUND5SZh_g4 3261
reax/metrics/evaluation.py sha256=JUojtzVQGOgKFyjjBfl5wLBhBYEkDDApYLjcyTmFZVY 1555
reax/metrics/metric.py sha256=G4lbMRtHRKF7j9vurY5ekTnPakh4HIMhx1r3wkAOJU0 3459
reax/metrics/regression.py sha256=yc2stqaHHk29bvkV7GD0xLc08avxylorvoRNXD6jDqw 3107
reax/metrics/utils.py sha256=SvpeaNvQW1UTwbk2dUdqbXe5CCOyDJXCrO3f2NOmnks 5050
reax/strategies/__init__.py sha256=f8VlAkCgRyhtdmfeJDd074BVe9jkF_mk2U-vN2JJw9g 130
reax/strategies/_single_device.py sha256=8gxjHAgdnrErnT0v9m_OQBMMAyaTxMSj58r85rbuyxM 447
reax/strategies/_strategies.py sha256=dS1vsFq-MsrQuX0CHbAIWgBWU7xB3ee1V_XbG3zhnK8 352
reax/utils/__init__.py sha256=bw_X62Hmx-7326Xxp8mcJ_1cS4FlMnoPlCdTsSJI0wk 87
reax/utils/arrays.py sha256=Lkg6_hNCcd6Kje0r9zD2F-vGOK9_Rt2PNECg5ACXtGw 222
reax/utils/containers.py sha256=vJnLW3S3Pagj7vE7xIogE0El3w0DhNOtVCVnTpLCdIg 1726
reax/utils/events.py sha256=XG11YG5yKgnvEPVWFofTZ2Xh7XRLz9Uz_d00-y_GBcw 1016
reax/utils/rngs.py sha256=ZBl1xf5N-lWm64XXmpFc65dRQI6xkK5faz36RwjRw6U 250
reax-0.2.0.dist-info/LICENSE.txt sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg 7652
reax-0.2.0.dist-info/WHEEL sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4 81
reax-0.2.0.dist-info/METADATA sha256=4jYoL2o6InSvx9TkItyi9x8hoCzZuWoN6nciKYjBglk 1858
reax-0.2.0.dist-info/RECORD — —