jax-tpu-embedding

PyPI page
Author: JAX SC Authors
Summary: JAX SparseCore API
Latest version: 0.2.0
Required dependencies: absl-py | einops | etils | flax | jax | numpy | portpicker

Downloads last day: 68
Downloads last week: 482
Downloads last month: 1,856
7-day smoothing