jax-tpu-embedding

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

Downloads last day: 91
Downloads last week: 1,058
Downloads last month: 1,665