Metadata-Version: 2.1
Name: onsagernet
Version: 0.1.0
Summary: 
Author: Qianxiao Li
Requires-Python: >=3.12,<4.0
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Provides-Extra: cpu
Provides-Extra: cuda
Provides-Extra: examples
Requires-Dist: datasets (>=3.0.0,<4.0.0)
Requires-Dist: jax (>=0.4.31,<0.5.0)
Requires-Dist: jax-cpu[cpu] (>=0.4.31,<0.5.0) ; extra == "cpu"
Requires-Dist: jax-cuda[cuda-12] (>=0.4.31,<0.5.0) ; extra == "cuda"
Requires-Dist: ml-dtypes (==0.4.0)
Requires-Dist: optax (>=0.2.3,<0.3.0)
Requires-Dist: pdoc (>=14.7.0,<15.0.0)
Requires-Dist: tqdm (>=4.66.5,<5.0.0)
