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:
13,102
Downloads last week:
88,470
Downloads last month:
316,439