jax-tpu-embedding

PyPI page
Home page
Author: None
Summary: JAX SparseCore API
Latest version: 0.1.0
Required dependencies: absl-py | einops | flax | jax | numpy | portpicker

Downloads last day: 215
Downloads last week: 1,231
Downloads last month: 8,266