jax_tpu_embedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax_tpu_embedding/prebuilt.cpython-313-x86_64-linux-gnu.so,sha256=HVlzS1ScKgewXqr9V6rrB8YpUYyiAPjjYMKOwFdipc4,14920
jax_tpu_embedding/sparsecore/__init__.py,sha256=2qBIWLu-pXNtaN9RRnR51KR1YBWzRFkQ6iZRtr5UL7M,698
jax_tpu_embedding/sparsecore/version.py,sha256=6RLbsz8z4O4cFRKo4COuyo0zQEAruN5ZctNci9fJPO0,415
jax_tpu_embedding/sparsecore/lib/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/auto_pipelining/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax_tpu_embedding/sparsecore/lib/auto_pipelining/auto_pipelining.py,sha256=lSnEGQ4kHxrl9T4Muo1JkUdfixMi6JsvuhUFshs-fBw,8541
jax_tpu_embedding/sparsecore/lib/auto_pipelining/decompose.py,sha256=UtqQ8ysNEIqXkh0XZ2RRW-PYucA2KUTFSr3ipgsQKDo,12722
jax_tpu_embedding/sparsecore/lib/auto_pipelining/preprocess.py,sha256=XDVnIpE6JfzNGpTECi7bQIkVAWwrpuTmmmp5BozfrmU,5401
jax_tpu_embedding/sparsecore/lib/auto_pipelining/utils.py,sha256=PPO7yfHdSpNlVPlRRHBlvBjKVwZRlx8PJTOfkO6jdsY,3416
jax_tpu_embedding/sparsecore/lib/core/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/core/constants.py,sha256=uWpZH1c4MIzU6UD-9Mle7SaJGwp8fc9ELJYDqT0pPyM,756
jax_tpu_embedding/sparsecore/lib/core/input_preprocessing.py,sha256=tEEI17VMssdE7JhVjPSbXNWeaLsf8ui-HOnOOAvEKF4,9528
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_core.so,sha256=w17tuG4-OYeGUWXJ2DE19sDuBC1NQO9iMYlEda_ETc0,164280
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_threads.so,sha256=80aJpVuqpsh4xYgEzkQJxAkgzKtPzwb30jq8R4C2a9E,9648
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_util.so,sha256=Flt2R6x0hRUNhB3hcgbHt-5Ror6D88v5bMQg4eR5oNw,21848
jax_tpu_embedding/sparsecore/lib/core/pybind_input_preprocessing.so,sha256=_W0yPKGsNIoNnvrhbe-TSz_R0t9FwDUEUJ8t7FJfW1Y,11199224
jax_tpu_embedding/sparsecore/lib/core/primitives/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/core/primitives/optimizers_computation.py,sha256=ELBgTGuXQG6rJrWLBCeUq2a6liB9AIP4sqGPa1P7-x8,4476
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_csr.py,sha256=3SXUQko8JsGpolvLD9iEeyuSXC9yRS8tLsjirEglPVk,6021
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad.py,sha256=RdYpUjbhfkRw4QVd8h4jTYzRVyLaO5csd1fA6Xy33eM,8540
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad_momentum.py,sha256=tE-9LDTaWa3Esi2XVf8i5WcLKiLrR-m4WgawM8xtKlE,11919
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adam.py,sha256=pZvEjeeyuS2WkqRy4yvVAglzLrc_xcfuwmN28nfbfMA,12620
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_ftrl.py,sha256=pv9ZeLaGNASSUikLR0IIFHls977MBBYFj9efwzm_1cQ,12368
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_laprop.py,sha256=y37sFftTIPRdDuHMcrCdxHGB0_wF7JnQXeEwJ1fI79Q,12712
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_sgd.py,sha256=XaQNxqFP0iZCtmqyQz15tk7BY5ZrL2HCGOz_P_HBTfg,7202
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_optimizer_grad.py,sha256=3knbkJ7--6ULWvIDrhgXHXh3HC-02S6xYI0Tfq9O1Yc,10224
jax_tpu_embedding/sparsecore/lib/core/primitives/utils.py,sha256=-_1MVaLSpu58CiAjx0TAsoDEtQGca1xs-BOG7KEMeJo,3592
jax_tpu_embedding/sparsecore/lib/fdo/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/fdo/fdo_client.py,sha256=nJHb1stdbBWbASQlIKY0I671Ebyz34gH00L3V4x200c,3058
jax_tpu_embedding/sparsecore/lib/fdo/file_fdo_client.py,sha256=2M5s7-To3T585cay7sm0tuDranD2QrdgKt2mulVURQM,7699
jax_tpu_embedding/sparsecore/lib/flax/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/flax/embed.py,sha256=5a6zIvSNm1UJNRI6dA6n56tuPEczXxnfzWVFsko_C90,9170
jax_tpu_embedding/sparsecore/lib/flax/embed_optimizer.py,sha256=ReAg5xlzen_xJ4orZQN8QwrH4dHdiODdVeb84mMk8A0,3042
jax_tpu_embedding/sparsecore/lib/nn/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/nn/embedding.py,sha256=Hs_7cdv9CcRnhW0H77RKd180Fy_psM2cMkt1QJsyiPw,57799
jax_tpu_embedding/sparsecore/lib/nn/embedding_spec.py,sha256=eghjpqU24OKWCDE32ihR8Xl-xXvU4ha8V4ZQGAgaUsU,27049
jax_tpu_embedding/sparsecore/lib/nn/table_stacking.py,sha256=0QKg8cvlYXCfHpkBWRnnN2itbySEYODX9LOwd-sJNKo,40405
jax_tpu_embedding/sparsecore/lib/proto/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/proto/embedding_spec_pb2.py,sha256=Pk5tueq7ItMa4qF8P0uddYVW6atkNe0NubxXe_Nww3s,3383
jax_tpu_embedding/sparsecore/utils/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/utils/utils.py,sha256=tsus1EWu0RllgUVmlTOoVPfzJhhWaS_b7dWXIwYBS0M,1684
jax_tpu_embedding-0.1.0.dev20251016.dist-info/METADATA,sha256=fcRPkz7pIYWdSQjP6XVCOD93QmQUVmc29bFYwAMrEn8,14120
jax_tpu_embedding-0.1.0.dev20251016.dist-info/WHEEL,sha256=WDfD3kSJ3BuT4YMWIyAY-9no4dOrDSu0e9jfkjodjXU,113
jax_tpu_embedding-0.1.0.dev20251016.dist-info/top_level.txt,sha256=YzWDDKrd9_GgdRp8EplkirWEw34u_Z7Fxffd1Fdb4OE,18
jax_tpu_embedding-0.1.0.dev20251016.dist-info/RECORD,,
jax_tpu_embedding-0.1.0.dev20251016.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
