mingfeisun/kfac : jax
Latest release: about 3 years ago - 168 downloads - 80 dependencies
JSONDependencies
- ;platform_system Darwin
- ;platform_system Linux
- ;platform_system Windows
- Markdown 3.4.4
- MarkupSafe 2.1.3
- PyYAML 6.0.1
- Werkzeug 2.3.6
- absl-py 1.4.0
- array-record 0.4.0
- astunparse 1.6.3
- cachetools 5.3.1
- certifi 2023.7.22
- charset-normalizer 3.2.0
- chex 0.1.7
- click 8.1.6
- cloudpickle 2.2.1
- contextlib2 21.6.0
- decorator 5.1.1
- distrax 0.1.3
- dm-haiku 0.0.10
- dm-tree 0.1.8
- etils 1.3.0
- flatbuffers 23.5.26
- gast 0.4.0
- google-auth-oauthlib 1.0.0
- google-auth 2.22.0
- google-pasta 0.2.0
- googleapis-common-protos 1.59.1
- grpcio 1.56.2
- h5py 3.9.0
- idna 3.4
- immutabledict 3.0.0
- importlib-metadata 6.8.0
- importlib-resources 6.0.0
- jax 0.4.13
- jaxlib 0.4.13
- jaxline 0.0.5
- jmp 0.0.4
- keras 2.13.1
- libclang 16.0.6
- ml-collections 0.1.1
- ml-dtypes 0.2.0
- numpy 1.24.4
- oauthlib 3.2.2
- opt-einsum 3.3.0
- optax 0.1.7
- packaging 23.1
- pip 20.0.2
- platform_machine aarch64
- promise 2.3
- protobuf 4.23.4
- psutil 5.9.5
- pyasn1-modules 0.3.0
- pyasn1 0.5.0
- requests-oauthlib 1.3.1
- requests 2.31.0
- rsa 4.9
- scipy 1.10.1
- setuptools 45.2.0
- six 1.16.0
- tabulate 0.9.0
- tensorboard-data-server 0.7.1
- tensorboard 2.13.0
- tensorflow-datasets 4.9.2
- tensorflow-estimator 2.13.0
- tensorflow-io-gcs-filesystem 0.32.0
- tensorflow-metadata 1.13.1
- tensorflow-probability 0.20.1
- tensorflow 2.13.0
- termcolor 2.3.0
- toml 0.10.2
- toolz 0.12.0
- tqdm 4.65.0
- typing_extensions 4.7.1
- urllib3 2.0.4
- wheel 0.34.2
- wrapt 1.15.0
- zipp 3.16.2