JAX SparseCore API

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for adityagupta1089 from gravatar.com adityagupta1089 Avatar for cantonios from gravatar.com cantonios

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License Expression: Apache-2.0
    SPDX License Expression
  • Author: JAX SC Authors
  • Tags jax , tpu , embedding , sparsecore , machine learning , sparse tensor , distributed computing
  • Requires: Python >=3.10
Classifiers
Report project as malware

Project description

jax_tpu_embedding

Unittests

Usage instructions coming soon!

This is not an officially supported Google product.

Project details

Verified details

These details have been verified by PyPI
Maintainers
Avatar for adityagupta1089 from gravatar.com adityagupta1089 Avatar for cantonios from gravatar.com cantonios

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License Expression: Apache-2.0
    SPDX License Expression
  • Author: JAX SC Authors
  • Tags jax , tpu , embedding , sparsecore , machine learning , sparse tensor , distributed computing
  • Requires: Python >=3.10
Classifiers

Release history Release notifications | RSS feed

0.1.0

This version

0.1.0.dev20260515 pre-release

0.1.0.dev20260503 pre-release

0.1.0.dev20260312 pre-release

0.1.0.dev20251208 pre-release

0.1.0.dev20251016 pre-release

0.1.0.dev20250915 pre-release

0.1.0.dev20250903 pre-release

0.1.0.dev20250618 pre-release

0.1.0.dev20250617 pre-release

0.1.0.dev20250520 pre-release

0.1.0.dev20250516 pre-release

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

jax_tpu_embedding-0.1.0.dev20260515.tar.gz (4.0 MB view details)

Uploaded Source

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

jax_tpu_embedding-0.1.0.dev20260515-cp314-cp314-manylinux_2_31_x86_64.whl (4.1 MB view details)

Uploaded CPython 3.14manylinux: glibc 2.31+ x86-64

jax_tpu_embedding-0.1.0.dev20260515-cp313-cp313-manylinux_2_31_x86_64.whl (4.1 MB view details)

Uploaded CPython 3.13manylinux: glibc 2.31+ x86-64

jax_tpu_embedding-0.1.0.dev20260515-cp312-cp312-manylinux_2_31_x86_64.whl (4.1 MB view details)

Uploaded CPython 3.12manylinux: glibc 2.31+ x86-64

jax_tpu_embedding-0.1.0.dev20260515-cp311-cp311-manylinux_2_31_x86_64.whl (4.1 MB view details)

Uploaded CPython 3.11manylinux: glibc 2.31+ x86-64

jax_tpu_embedding-0.1.0.dev20260515-cp310-cp310-manylinux_2_31_x86_64.whl (4.1 MB view details)

Uploaded CPython 3.10manylinux: glibc 2.31+ x86-64

File details

Details for the file jax_tpu_embedding-0.1.0.dev20260515.tar.gz.

File metadata

File hashes

Hashes for jax_tpu_embedding-0.1.0.dev20260515.tar.gz
Algorithm Hash digest
SHA256 3f0de3e89a19cb5ae26746996ddbc85ee4badf746b3b2cba0eb1191c960496aa
MD5 e3af5708ec97589448daec3c434211e5
BLAKE2b-256 ade0efb9628d123e832fc804d7273693396e1280b32c9ac9fea78b1a8e8f7465

See more details on using hashes here.

File details

Details for the file jax_tpu_embedding-0.1.0.dev20260515-cp314-cp314-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for jax_tpu_embedding-0.1.0.dev20260515-cp314-cp314-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 3ad6e3fe11a4a4cb55fb7a59487cfe39430c409983837a66ef0f5b60987860ee
MD5 65bf6aaf93d23b34fa812c517f4df674
BLAKE2b-256 18d188acd4f2195c05933e4a32713451ab84e64bcbf6b73da899d38be5818109

See more details on using hashes here.

File details

Details for the file jax_tpu_embedding-0.1.0.dev20260515-cp313-cp313-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for jax_tpu_embedding-0.1.0.dev20260515-cp313-cp313-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 db03ec05f9d999d97dafa9dfc7c108a5ef54de1af99d10f878881672f644aae1
MD5 87366af9d10047993cf2ef299def9302
BLAKE2b-256 b9d18b03039c0ea67c412b8e42d9f4b9c091984b89851c2a18d8733aa506ab49

See more details on using hashes here.

File details

Details for the file jax_tpu_embedding-0.1.0.dev20260515-cp312-cp312-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for jax_tpu_embedding-0.1.0.dev20260515-cp312-cp312-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 957462c299118fc1c51d108a748c3e53d10a39b37978f6a16fc316235c63b4a5
MD5 ee0f3c2ce2ae86e643bd5bfb5a2ca272
BLAKE2b-256 cc951cfcd4c82dc18b78df4a33eb13fac54437f0cebb03b4fa1d060487400e04

See more details on using hashes here.

File details

Details for the file jax_tpu_embedding-0.1.0.dev20260515-cp311-cp311-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for jax_tpu_embedding-0.1.0.dev20260515-cp311-cp311-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 b3cd4732f775a2d7f57fbf360fc6f77f4c94990233085ffdb978b05e89bc4f2b
MD5 c9f14636e393129c4378efca5b4c3bfa
BLAKE2b-256 3a4372c721099db1ba2ed790a4ef6cfe27434b85d420801bff083278dfb902ca

See more details on using hashes here.

File details

Details for the file jax_tpu_embedding-0.1.0.dev20260515-cp310-cp310-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for jax_tpu_embedding-0.1.0.dev20260515-cp310-cp310-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 1f924dc44341d356006c02241ed3df470eca2efc1cd75e2f9fe372e06e1854af
MD5 f64bbeb8c3fe6cb417b1df2c6e7d8691
BLAKE2b-256 4f3391b01f4abc21b0b60aabe657a85b1401c5893216ab5297c51f9d896851b5

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page