Wheelodex » Projects » reax

reax

View on PyPI — Reverse Dependencies (0)

0.4.0 reax-0.4.0-py3-none-any.whl

Wheel Details

Project: reax
Version: 0.4.0
Filename: reax-0.4.0-py3-none-any.whl
Download: [link]
Size: 115162
MD5: 042ab424684531f0158c005ae23a316c
SHA256: 4325075ebf8f260a81e72f5f31d220c85ab14aaedf3f3a9aee98f4cbd350284e
Uploaded: 2025-02-20 13:25:30 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.1
Name: reax
Version: 0.4.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: 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=wFUBkiRCIyR2UBTqGI_dnQUqvVWy9bnW7oKtrCnS5DQ 1119
reax/_evaluation.py sha256=Wkq0xsOKsr77Ci3c8hsZ5wN8OZuZ7NHKgi41qKxzp68 1937
reax/_module_hooks.py sha256=MYdX-cZlkH8FtpB5dme77U7DkmppqHQNoe5tPQ_648g 5212
reax/exceptions.py sha256=l2d0Z0wu91TfJl_smSKQXME-9Nglxajn4xbCE5nIo-4 123
reax/hooks.py sha256=TZo-AYzGwO11NwdITdwp-fuDMmod5Ku2tCknHVIsRfo 5580
reax/keys.py sha256=5lPOT7YdTGOff9JQsgzae83DY0waSVfgbfToBBQnyqc 274
reax/modules.py sha256=mEMQvJGTUgrQgRB013i0sEnFLz7XHh-NH432KOIDs-c 7822
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=_MQGro6WuHw-2gO92Cb2_fqFwsWfCVQfQdFSxjmaKAY 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=bQjvEuVuAXqNkmDtI5cmtPVm5C3B5YMlQqSZQjmzUPc 4283
reax/metrics/_registry.py sha256=c-USh1OJtFlNSftSuWeO_T15Nj5s7kUi1WeYb89pLnI 1246
reax/metrics/aggregation.py sha256=AE9XctKGN_DgcxjQMKOU4VB-UV54wXayhzmRbVGcmis 5070
reax/metrics/collections.py sha256=UC_s27HhVztuzjI2lHsLhEta2priI8z127b4F7VaHlw 3467
reax/metrics/regression.py sha256=ctdWpw4ZzNiEKkOlL0zdlqQhyY25u2Ass-v5DEwX4RQ 3462
reax/metrics/utils.py sha256=MW2LvvI-NVYWDX0Y4ABViuqXnBF_55gGErxK8wL6Mfk 6939
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=icnXOFqAOhX9v2E8BNipoWOs5cxW49x4CFodL6QMwEQ 3609
reax/stages/fit.py sha256=k53xQ1oKgMPUD441uqf3P9xXWmHBJkbi8hdNn5Gf2AI 12059
reax/stages/predict.py sha256=OXvN2e-q5KlEtQyoOLyNLmXj5BlGmke30pB2BXxVPFA 1268
reax/stages/stages.py sha256=BfoFrCjzd930bKJA4Z91MDLFJ86aoGmKLUHJ2wyQE1g 13643
reax/stages/test.py sha256=mqeIsuVT3c02cZQLEr-9PiHJmmurJ-1_0PxgHhxuGvo 1267
reax/stages/train.py sha256=5HkXK9QgfLHPneBC3WReiyYoIok86Pq1LNVl9p8YUWw 5586
reax/stages/validation.py sha256=p0OwR5H8U6MOFqGoWUwRtJCPU7gHopN2OtaMltaaUsM 1487
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=JYfd_9wKPEk1TPrplvWyPcgpiO9QUfKunZSytbUKrIM 25791
reax/utils/__init__.py sha256=TNIt41sn2ZFKnfTJc26dRkcp0ZHIBgLOV_iX_KXdJ2w 131
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/grads.py sha256=978-xYLB1cFV29jcWSCqxkMnHm-vg5qfDMY6aOo9gO4 3342
reax/utils/rngs.py sha256=VsSAxjJFKVsNsTNmlrXHHczxaTQml5kPdM9Kn9lH4c0 277
reax-0.4.0.dist-info/LICENSE.txt sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg 7652
reax-0.4.0.dist-info/WHEEL sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4 81
reax-0.4.0.dist-info/METADATA sha256=yIMa2bZCKYmIxggiQPzXybCcREUE_oy5maHdMYY9Y30 6237
reax-0.4.0.dist-info/RECORD — —