Metadata-Version: 2.4
Name: jax_tpu_embedding
Version: 0.1.0.dev20260312
Summary: JAX SparseCore API
Author-email: JAX SC Authors <jax-sc-dev@google.com>
License-Expression: Apache-2.0
Project-URL: homepage, https://github.com/jax-ml/jax-tpu-embedding
Project-URL: repository, https://github.com/jax-ml/jax-tpu-embedding
Project-URL: issue_tracker, https://github.com/jax-ml/jax-tpu-embedding/issues
Keywords: jax,tpu,embedding,sparsecore,machine learning,sparse tensor,distributed computing
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: absl-py
Requires-Dist: einops
Requires-Dist: flax
Requires-Dist: jax
Requires-Dist: numpy
Requires-Dist: portpicker
Dynamic: license-file

# jax_tpu_embedding

[![Unittests](https://github.com/jax-ml/jax-tpu-embedding/actions/workflows/build_and_test.yml/badge.svg)](https://github.com/jax-ml/jax-tpu-embedding/actions/workflows/build_and_test.yml)

Usage instructions coming soon!

*This is not an officially supported Google product.*
