PyPI page
Home page
Author:
DeepMind
License:
Apache 2.0
Summary:
DSL-based reshaping library for JAX and other frameworks
Latest version:
1.0
Required dependencies:
absl-py
|
dataclasses
|
numpy
Optional dependencies:
jax
|
jaxlib
|
tensorflow
|
tensorflow-gpu
Downloads last day:
15,281
Downloads last week:
135,971
Downloads last month:
447,670