flax-pilot

View on PyPIReverse Dependencies (0)

0.2.3 flax_pilot-0.2.3-py3-none-any.whl

Wheel Details

Project: flax-pilot
Version: 0.2.3
Filename: flax_pilot-0.2.3-py3-none-any.whl
Download: [link]
Size: 12856
MD5: b2186f90fc75f2784493942cd8dc055e
SHA256: 40b7fe9d037018f4a835bb76293e4264d9f8879ad00a8979dd69514f1b795c38
Uploaded: 2024-12-25 13:50:19 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: flax-pilot
Version: 0.2.3
Summary: A Simplistic trainer for Flax
Author: Nithish M
Author-Email: nithishm2206[at]gmail.com
Home-Page: https://github.com/NITHISHM2410/flax-pilot
Classifier: Programming Language :: Python :: 3.10
Classifier: License :: OSI Approved :: MIT License
Requires-Python: >=3.10
Requires-Dist: jax[cuda12] (==0.4.37)
Requires-Dist: flax (>=0.10.2)
Requires-Dist: optax (>=0.2.4)
Requires-Dist: numpy
Requires-Dist: tqdm
Requires-Dist: mergedeep
Requires-Dist: orbax-checkpoint (>=0.5.20)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 3602 characters]

WHEEL

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

RECORD

Path Digest Size
fpilot/__init__.py sha256=8XwaBx3Aof9UiehcYG8-OLZl6Jqy804-j8ic01cD3MY 82
fpilot/training/__init__.py sha256=X7JbOm3x6R8qb5_bUko2DME_mjZq9EMeBIepkXiTY9g 43
fpilot/training/trainer.py sha256=-zWkEvafrM330i3rUzOWbtuTAOk_DRQq2WcYgYo9-44 19785
fpilot/utils/__init__.py sha256=rrAdwtXIPe9PZqkvcMHrWmHkUxK3bHqW_60YT8xd1YY 105
fpilot/utils/checkpoint_utils.py sha256=QF39deBY6J1CrMIlzyxEnGYzzVvtmUeGJBCALz8um5I 817
fpilot/utils/common_utils.py sha256=7PUl2RbBS7lWPeLmaCrGXEvHR218B3GfajCEyB4V85s 469
fpilot/utils/opt_utils.py sha256=JO6LljlyZhGdw_NmrzH8jy4S4A_DQhR8BGS51utMNVQ 1200
fpilot/utils/tracker_utils.py sha256=Qbk-pHg7x8EbOwgRgSN8Y0PMs6J8CGnWwEyzmAILLJ0 5530
fpilot/utils/trainer_utils.py sha256=JwuWr5PO39htCfkG3azLXrnIEOUg6qn1VtXnCVIY5SI 459
flax_pilot-0.2.3.dist-info/LICENSE sha256=d-cFhq0g53th2fafGXk1VOzQDWqpGlyWiPutuKqgLas 1087
flax_pilot-0.2.3.dist-info/METADATA sha256=EEScpmdSIegPx14NYHJwiueWH2kwYr6WiKNHF7EArm4 4321
flax_pilot-0.2.3.dist-info/WHEEL sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ 91
flax_pilot-0.2.3.dist-info/top_level.txt sha256=91_HUIwm9IWyY-mJMNVYpMCmSdQkDrPkBwahOwBGAZQ 7
flax_pilot-0.2.3.dist-info/RECORD

top_level.txt

fpilot