Pre-release

This release is a pre-release and may not be stable for production use.

jax_tpu_embedding

Unittests

Usage instructions coming soon!

This is not an officially supported Google product.

Project links

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.2.0)

2 maintainers

PyPI data
Data sourced directly from PyPI's database.
Avatar for adityagupta1089 from gravatar.com adityagupta1089 Avatar for cantonios from gravatar.com cantonios

Credits

Author: JAX SC Authors

License expression

Apache-2.0
View SPDX License List

Requires

Python >=3.10

Tags

jax tpu embedding sparsecore machine learning sparse tensor distributed computing

Classifiers

Report project as malware

Metadata

Project links

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.2.0)

2 maintainers

PyPI data
Data sourced directly from PyPI's database.
Avatar for adityagupta1089 from gravatar.com adityagupta1089 Avatar for cantonios from gravatar.com cantonios

Credits

Author: JAX SC Authors

License expression

Apache-2.0
View SPDX License List

Requires

Python >=3.10

Tags

jax tpu embedding sparsecore machine learning sparse tensor distributed computing

Classifiers

Report project as malware

Release files for jax-tpu-embedding 0.1.0.dev20260503

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for jax-tpu-embedding 0.1.0.dev20260503
File Size Uploaded
jax_tpu_embedding-0.1.0.dev20260503.tar.gz 4.0 MB Details

Built distributions (wheels)

Table of built distributions (wheels) for jax-tpu-embedding 0.1.0.dev20260503
File
jax_tpu_embedding-0.1.0.dev20260503-cp313-cp313-manylinux_2_31_x86_64.whl 4.1 MB CPython 3.13 CPython 3.13 Linux glibc 2.31+ x86-64 Details
jax_tpu_embedding-0.1.0.dev20260503-cp312-cp312-manylinux_2_31_x86_64.whl 4.1 MB CPython 3.12 CPython 3.12 Linux glibc 2.31+ x86-64 Details
jax_tpu_embedding-0.1.0.dev20260503-cp311-cp311-manylinux_2_31_x86_64.whl 4.1 MB CPython 3.11 CPython 3.11 Linux glibc 2.31+ x86-64 Details
jax_tpu_embedding-0.1.0.dev20260503-cp310-cp310-manylinux_2_31_x86_64.whl 4.1 MB CPython 3.10 CPython 3.10 Linux glibc 2.31+ x86-64 Details

Total release size: 20.5 MB

Release files / jax_tpu_embedding-0.1.0.dev20260503.tar.gz

Download URL jax_tpu_embedding-0.1.0.dev20260503.tar.gz
Size 4.0 MB
Tags Source
SHA-256 checksum
How to use checksums 		4b2cad90d6ea46c64103ee910c4463bb433c7c550eb5f80f0647f6ac5fb316d9
BLAKE2b-256 checksum
How to use checksums 		c1ac0c0276b05fcd1dabc26907c2f168b9cbb186d28a8b0c7f1692b3a8a1c124
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/6.1.0 CPython/3.10.12

Release files / jax_tpu_embedding-0.1.0.dev20260503-cp313-cp313-manylinux_2_31_x86_64.whl

Download URL jax_tpu_embedding-0.1.0.dev20260503-cp313-cp313-manylinux_2_31_x86_64.whl
Size 4.1 MB
Tags CPython 3.13 Linux glibc 2.31+ x86-64
SHA-256 checksum
How to use checksums 		2b11519d2156d1072981d3a1f6d5037973eda9178b34ca7fad885f5f30352930
BLAKE2b-256 checksum
How to use checksums 		c7fd6ac259d7052676ed7082ce7d838e1a359aad45f9d3a777891330d0b4e685
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/6.1.0 CPython/3.10.12

Release files / jax_tpu_embedding-0.1.0.dev20260503-cp312-cp312-manylinux_2_31_x86_64.whl

Download URL jax_tpu_embedding-0.1.0.dev20260503-cp312-cp312-manylinux_2_31_x86_64.whl
Size 4.1 MB
Tags CPython 3.12 Linux glibc 2.31+ x86-64
SHA-256 checksum
How to use checksums 		7a5a6f9f860e358538fabb763b3d8d10b7b0b4097792d6f176d602f86e913041
BLAKE2b-256 checksum
How to use checksums 		d3751ac800b4af2efaeb2130ced94d9aaf13c3ae25819ff64da5c01a75e56041
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/6.1.0 CPython/3.10.12

Release files / jax_tpu_embedding-0.1.0.dev20260503-cp311-cp311-manylinux_2_31_x86_64.whl

Download URL jax_tpu_embedding-0.1.0.dev20260503-cp311-cp311-manylinux_2_31_x86_64.whl
Size 4.1 MB
Tags CPython 3.11 Linux glibc 2.31+ x86-64
SHA-256 checksum
How to use checksums 		5498843aaf525a066c78a2d53749e442f9db3ba101f5335dfd5e5ce304c1d462
BLAKE2b-256 checksum
How to use checksums 		40edf8718a09bb995d0fe95b8ce9fd96f209cbc18813f299181d9753dd9fc1aa
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/6.1.0 CPython/3.10.12

Release files / jax_tpu_embedding-0.1.0.dev20260503-cp310-cp310-manylinux_2_31_x86_64.whl

Download URL jax_tpu_embedding-0.1.0.dev20260503-cp310-cp310-manylinux_2_31_x86_64.whl
Size 4.1 MB
Tags CPython 3.10 Linux glibc 2.31+ x86-64
SHA-256 checksum
How to use checksums 		08bb8fa9b67714f20a2b3ac5cf0987b81fbd951da1e2598e5e13bf30b70e46f5
BLAKE2b-256 checksum
How to use checksums 		c058ab1ca65eb9cd8a8a20fee3ab661244d807548e8d22ce8249f2eb87564a12
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing? 		No
Uploaded via twine/6.1.0 CPython/3.10.12
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page