Wheelodex » Projects » reax » reax-0.3.1-py3-none-any.whl

reax

View on PyPI — Reverse Dependencies (0)

0.3.1 reax-0.3.1-py3-none-any.whl

Wheel Details

Project: reax
Version: 0.3.1
Filename: reax-0.3.1-py3-none-any.whl
Download: [link]
Size: 111971
MD5: eef530a0f8652f3f23f404e17f8aea89
SHA256: 68318043c72e67a40a5038470fa4de55e91db00d4af9e917b191c5da2d4b18bc
Uploaded: 2025-01-24 12:00:38 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.1
Name: reax
Version: 0.3.1
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: einops
Requires-Dist: clu
Requires-Dist: flax
Requires-Dist: fsspec
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: jaxtyping
Requires-Dist: lightning-utilities (>=0.10.0)
Requires-Dist: optax
Requires-Dist: orbax-checkpoint
Requires-Dist: pytray
Requires-Dist: equinox
Requires-Dist: tensorboardX
Requires-Dist: tqdm
Requires-Dist: typing-extensions
Requires-Dist: black; extra == "dev"
Requires-Dist: cloudpickle; extra == "dev"
Requires-Dist: flit; extra == "dev"
Requires-Dist: ipython; extra == "dev"
Requires-Dist: mlflow; extra == "dev"
Requires-Dist: pandas; 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: docutils; extra == "docs"
Requires-Dist: jupyter; extra == "docs"
Requires-Dist: nbsphinx; extra == "docs"
Requires-Dist: pandoc; extra == "docs"
Requires-Dist: scikit-learn; 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: 4141 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=3lDpvh0xBseIExHELwsKMHBlXmRDkjYLNMKCv7nu_R4 1119
reax/_evaluation.py sha256=wi0Ne_D6X9_vHo0LUTJI0z3iwIDrX2uqDMBabT65z6g 1725
reax/_module_hooks.py sha256=10_OmOyXZ8Rk8RSnnRkO0Zt_jJ_Tr2qE6dVxntvqgCo 4680
reax/exceptions.py sha256=l2d0Z0wu91TfJl_smSKQXME-9Nglxajn4xbCE5nIo-4 123
reax/hooks.py sha256=TZo-AYzGwO11NwdITdwp-fuDMmod5Ku2tCknHVIsRfo 5580
reax/keys.py sha256=5LbKw7-OEeMYuYVPMYqc7oo7zeQjtWuY2XuFKeuDHws 95
reax/modules.py sha256=wA8gBnFRD57BLxcdghgyJRPvv7OSsDI8vjO690eUSDw 5500
reax/optimizers.py sha256=Yv_HGUbzTaXkURXVGEf8IYTrG3slYUSKGub7lor0xUo 2159
reax/results.py sha256=XE0ltC595057UdwnjBwZqlw672YMSGPvHGK7VWCsSaI 3710
reax/saving.py sha256=Z2ktkeSNbifnRizMX9O7rrzlzGmIQZKYuN02P9SdQi0 3639
reax/typing.py sha256=vH7lmD3cOnN9qTz4BAy1RVDo9vrCa05TBRpMQ3PFGYY 245
reax/data/__init__.py sha256=45dzFccrpPJEBXTPoLrrK4ReJsw_4qAs5SGdfZ87NnY 397
reax/data/_loaders.py sha256=g_vGJ25Ryp0T2oqedJFkq-0soZbjGyHRmz228zVObgI 4203
reax/data/_types.py sha256=uwDRCbTxFc7MkVYsNeDbmAibmMg_y9uDcTzWkit-icU 361
reax/data/collate.py sha256=6rT9qcs79l5twxmcdt-_uDbWEpVGWYDoTojJfEbGlsI 5424
reax/data/data_loaders.py sha256=tLjGczKt7QUoOlLPkEV9rQFs4JN7dMWsK0u6QhUpnqg 1011
reax/data/data_modules.py sha256=sakl5VzR9aCA3kVvyMbeKmINT0TLquTyDuHqYiProgc 2817
reax/data/fetchers.py sha256=JANWswJOoCE5k_bDRgB53j3JUFviCTNcyi76FBBkFGo 3060
reax/data/samplers.py sha256=bGipwt_QDo8ZYCekWq6x2YsskigTCch9d4HmgQfg2oc 6737
reax/data/utils.py sha256=Wm6CblxoPEdWzUvBZZeGuEsrreRXPQXl0cQXXIi4EkM 3480
reax/demos/__init__.py sha256=_yCEfuY14QEGZABjW27ajwW5DxOhbzRZsESDBmGaGIw 93
reax/demos/boring_classes.py sha256=eBmsNaidn5BW-PzdNpCSWNpHqISmcSGWwAzOTZ66vUI 3844
reax/lightning/LICENSE sha256=pB4vrpkVrlYCj429C_J5lfGQdhPLzS2BphsBCtNOn-k 11349
reax/lightning/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
reax/lightning/rank_zero.py sha256=-US8XBaELkPkuS1r7C7CX50EypJeTwkHB1CJmG_1efg 2985
reax/listeners/__init__.py sha256=0_61VG-QcHuYsCjZN3wiUBShpS4O0Hn8FaDgX9-UNrc 373
reax/listeners/checkpointer.py sha256=GHAyYlXXmTjfoA23ZzE0FsRithEEdJWw-ffjQL5fi14 145
reax/listeners/early_stopping.py sha256=XtOSEsNA5DOMoVvD6Ri_siKWENBjQS_D1zWtuDyEg-Q 12983
reax/listeners/model_checkpoint.py sha256=bDDPP9azskD4yJ_Vb0CCG8-VFted5keeACTfn7QIPEo 29644
reax/listeners/utils.py sha256=AvGKkYMJXujcYsre_vs7BGo9P2Ix3uGGg9eIf5sHPII 338
reax/listeners/progress/__init__.py sha256=WQZO94Sl4vl7vAgZswMlH0t1xg0ERin2mumHcZNiwBA 155
reax/listeners/progress/progress_bar.py sha256=bLhLwcSHO6hKcXHuCKglPD2I870sRUdC4Oym7tNrx2g 610
reax/listeners/progress/tqdm_progress.py sha256=rxGBZFHP4YJ_6BPn241eMDWOWcbyQOVwjjXvBcj4-dg 12516
reax/loggers/__init__.py sha256=l1PuNLtMN6cJu9EmAAMJziGghjCAC2VkTWOpv_mBQm4 178
reax/loggers/_utils.py sha256=tlA4H0NxR7Hr2MefGKywc6mDB58qiAl-yzX6YQZLGoo 4713
reax/loggers/logger.py sha256=zi-QUIprSFRv6FY-cxehqq1JKziQVuERbQeMbnpCEFE 8179
reax/loggers/mlflow.py sha256=aS863S6uEUOpGjzd4q-ugyYkkT2U_cz4eV2SzpddK_E 17472
reax/loggers/pandas.py sha256=NACahqczlmsqpMBFXZww74Hhtyz1oqtypg7O9TCx10Q 8246
reax/loggers/tensorboard.py sha256=Zuo1a7MCnp0BfhG2nTOcato62tl9DtiPQIC4nvB49lQ 13321
reax/metrics/__init__.py sha256=1ac1OioZTA1wGCUdgU4a-cTCHjdVGoGxeRCrSBHqKqs 555
reax/metrics/_classification.py sha256=yiPwT9pTr3aDLsMeVFyZiN3nzxwhJbkLo8KvbiRTrZc 13621
reax/metrics/_globals.py sha256=uNuV8i0K9Ibb-aiUwynGMFdtRCX56gJab09Z6dXGH0w 621
reax/metrics/_metric.py sha256=L_WHMfF9Za3hOmZW5X4nFuYME7NY0Du7-vSKX1t_gLQ 3784
reax/metrics/_registry.py sha256=c-USh1OJtFlNSftSuWeO_T15Nj5s7kUi1WeYb89pLnI 1246
reax/metrics/aggregation.py sha256=8LQU3uLlXUSdpxaTKYkwSsZpfUjWyB-u9jeG5I_6xbM 5047
reax/metrics/collections.py sha256=UC_s27HhVztuzjI2lHsLhEta2priI8z127b4F7VaHlw 3467
reax/metrics/regression.py sha256=ctdWpw4ZzNiEKkOlL0zdlqQhyY25u2Ass-v5DEwX4RQ 3462
reax/metrics/utils.py sha256=TCAICHyvpHHonxJAutF87T-qqcOcZTNmuC-uKv1eaJ4 6232
reax/metrics/jm/__init__.py sha256=eKSafyoeL6KMVu8p6q5Rt6ciEVqZtzbRn18osDw2iXg 42
reax/metrics/jm/types.py sha256=rTGdRJmx-SVPXtofeG_5Qggy7hOe4BfvoO9CbWOLHq8 1474
reax/metrics/jm/utils.py sha256=63IacgGVv5Xi883Wsmn4DCCUFwmeP3ndrOdKbJmLK6g 36391
reax/stages/__init__.py sha256=WAJTinTgmXOOKntx5JAH5tnTIGZgxLMM3k1N1m6Gi5M 380
reax/stages/common.py sha256=eej0B9Eppjykbdxv2v1DHW3aLucRIrCP6-Kgj4UjEMs 3615
reax/stages/fit.py sha256=MzG9Jc194uUIJHLnaOGMuAfwF_AqMFdezIykSmLwWy4 11968
reax/stages/predict.py sha256=quPeeLjJJ-7wOZ99d8W6RQ-T1K3tWue6yaZD-CmbvPw 1247
reax/stages/stages.py sha256=KBDVa50VumVtw8rD2d9Z-hZLyuTz69fc-XeTfhgEsP8 13618
reax/stages/test.py sha256=u7ffAIA4YuxlT1e0twTNSrGklsoA-_cF61SB8ItD2TI 1236
reax/stages/train.py sha256=mYdgVI4uEMneW4iheGdYvM82A276ktQNc7DY-AW0_AI 5518
reax/stages/validation.py sha256=CkiVGZnkLDeiv78PdrpMmxkecwLePZVhJ4e78C3ApNo 1456
reax/strategies/__init__.py sha256=f8VlAkCgRyhtdmfeJDd074BVe9jkF_mk2U-vN2JJw9g 130
reax/strategies/_single_device.py sha256=2UxZD-R4nnspEUkmVawxrKbVWHeOzcL3GKMoSHmmWEQ 846
reax/strategies/_strategies.py sha256=cwMNgsgiz60AnxowktzX-X5ZLZ1tO-x4GuP7qOKw0qU 2431
reax/training/__init__.py sha256=hUh58vN-ojYOMgMeEMbaRSE20xj0ZhLa-YofJNdczl4 72
reax/training/_checkpoint_connector.py sha256=sbJXYSHakP7iRbiPkX64X5-nG5MoNIyd1OIxR5o7nbI 901
reax/training/_logger_connector.py sha256=nFYz95skZeHkpmLN3Hxhbilz8CminnCuuCvnUbKY15A 1754
reax/training/trainer.py sha256=KqM-pwlgmrbqXzPJYX68l_3aMwLinABLSabpCPd1XxE 25642
reax/utils/__init__.py sha256=bw_X62Hmx-7326Xxp8mcJ_1cS4FlMnoPlCdTsSJI0wk 87
reax/utils/arrays.py sha256=P-1sNI4QJKOFTVVkvLyv55VGE_CUQ3EEN8vOqiiIfUU 603
reax/utils/containers.py sha256=FzOmPLZwQPiEws_spCjt9UgBvVl_2U4ppvFWbywboYs 1996
reax/utils/events.py sha256=yi5QkLQicQTtLHPv0CoB3-vClp0-hs7atgTv875KaMg 1739
reax/utils/rngs.py sha256=VsSAxjJFKVsNsTNmlrXHHczxaTQml5kPdM9Kn9lH4c0 277
reax-0.3.1.dist-info/LICENSE.txt sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg 7652
reax-0.3.1.dist-info/WHEEL sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4 81
reax-0.3.1.dist-info/METADATA sha256=4MaPg8u7_bml0uwsqMtD5DyORfq0l-beszv3ybaLe4w 6237
reax-0.3.1.dist-info/RECORD — —