Metadata-Version: 2.1
Name: hijax
Version: 0.1.0
Summary: An experiment framework for Haiku and Jax
License: MIT
Author: Angus Turner
Author-email: angusturner27@gmail.com
Requires-Python: >=3.8,<3.11
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Requires-Dist: black (>=21.10b0,<22.0)
Requires-Dist: dm-haiku (>=0.0.5,<0.0.6)
Requires-Dist: hydra-core (>=1.1.1,<2.0.0)
Requires-Dist: jax (>=0.2.24,<0.3.0)
Requires-Dist: mypy (>=0.910,<0.911)
Requires-Dist: numpy (>=1.21.4,<2.0.0)
Requires-Dist: optax (>=0.0.9,<0.0.10)
Requires-Dist: torch (>=1.10.0,<2.0.0)
Requires-Dist: tqdm (>=4.62.3,<5.0.0)
Requires-Dist: wandb (>=0.12.6,<0.13.0)
