jax-ml/jax-tpu-embedding

PythonApache-2.0active

Health

High-performance large-scale embedding acceleration for JAX on Google TPU SparseCores.

Stars37

Forks5

Open Issues4

Contributors5

Last Push0d ago

Health Breakdown

Activity

Community

Maintenance

View on GitHub ↗Issues (4) ↗Pull Requests ↗

Should you contribute to jax-ml/jax-tpu-embedding?

jax-ml/jax-tpu-embedding has a FoundDev health score of 80/100, which puts it in the active-and-maintained tier. The maintainer team is shipping recently, issues are being closed, and a PR you open this week has a realistic chance of being reviewed.

Last push was 0 days ago — that signals an actively maintained project. New issues are likely to get a maintainer response within days. The project is written primarily in Python, so prior Python experience will shorten ramp-up.

Licensed under Apache-2.0, a standard OSI-approved license — safe to contribute to under normal employer IP policies.