xtrain

View on PyPIReverse Dependencies (1)

0.4.3 xtrain-0.4.3-py3-none-any.whl

Wheel Details

Project: xtrain
Version: 0.4.3
Filename: xtrain-0.4.3-py3-none-any.whl
Download: [link]
Size: 14379
MD5: 88aac34fbbe72b12cafa9b7796246043
SHA256: 5dba9a035af9c7355246305b2ecf5b43602768ae4e8553b983cc5c22cef928a3
Uploaded: 2024-12-18 18:04:42 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: xtrain
Version: 0.4.3
Summary: A Flax trainer
Author: Ji Yu
Author-Email: jyu[at]uchc.edu
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Python: >=3.10,<4.0
Requires-Dist: flax (>0.8)
Requires-Dist: jax[cuda12] (<0.5,>=0.4); extra == "cuda12"
Provides-Extra: cuda12
Description-Content-Type: text/markdown
[Description omitted; length: 1746 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.9.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
xtrain/__init__.py sha256=GAphzP2g9-JvCkioUcry4ntm5yxRT5YSaQzK1rTpkSA 190
xtrain/base_trainer.py sha256=CMePejm-IqSmyv_FjQedxmJLs6jpGl4u08xp3gcTX9A 15351
xtrain/closure.py sha256=H1WHqXgkSj03cCaED_RxUn-KRLhXkMsaL99gDJEs6As 5980
xtrain/loss.py sha256=4eaPvCC-jsNsklp5BmGlIeY_prkyQ7tdsYrBVdIW3Qk 1744
xtrain/strategy.py sha256=IE6H2vL50xZ6GdPfzJWRoW0KmBXp5OFeEl9IrO8003k 5534
xtrain/utils.py sha256=YbO1HO9KWNUz34aSoYAvT_qyKiV0d8wkVBjvhHT2G1U 8966
xtrain-0.4.3.dist-info/LICENSE sha256=VwqyKijqQvysv0b7dDxbvdWDwrApqXQYdy6iUm6zJM8 1062
xtrain-0.4.3.dist-info/METADATA sha256=zULwicFJNnozixYtUmbTL7mZMlxAhJ0qH4MkVhy20cE 2293
xtrain-0.4.3.dist-info/WHEEL sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs 88
xtrain-0.4.3.dist-info/RECORD