jax-tpu-embedding
High-performance large-scale embedding acceleration for JAX on Google TPU SparseCores.
View on Github
Activity
- Latest release
- 5d ago
- Total releases
- 2
- Cadence
- ~3 months
- Last 12 months
- 2
Reach
- Stars
- 37
Details
- License
- Apache-2.0
- First release
- Jun 11, 2026
Releases
| Version | Released | |
|---|---|---|
0.2.0
minor
| ||
0.1.0
initial
|