jax_tpu_embedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax_tpu_embedding/sparsecore/__init__.py,sha256=2qBIWLu-pXNtaN9RRnR51KR1YBWzRFkQ6iZRtr5UL7M,698
jax_tpu_embedding/sparsecore/version.py,sha256=Fv1whtNX_gikbOBMp_kp2KWd1qKRaYq2bKr3kg5C8fA,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=_y1xLcXti7kKhle-O46pDx2bXpoTjnp_Rw-Zu0CSm0s,8849
jax_tpu_embedding/sparsecore/lib/auto_pipelining/decompose.py,sha256=aKvNn1R5C2OPDinPE3TeHlNiCTTKI3XSBCUJZrEzzlc,12838
jax_tpu_embedding/sparsecore/lib/auto_pipelining/preprocess.py,sha256=Y0PBd0BqYN5iHZ3r2vxSQdLr3Ac1FNCKu6coyHEPicY,5341
jax_tpu_embedding/sparsecore/lib/auto_pipelining/utils.py,sha256=bqoh2uPTdguSGHOUfPnGe44Em5FNT7ym_Che8pkc0-s,4136
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=P0WxrEPRSeeU00Iri5utAtPc2pMYh_0Zh-dE_qg-uFw,9473
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_core.so,sha256=eVhtG26ZGmED81JyX_mqaUYEE26xqBXSv92ODzOsqj4,587744
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_threads.so,sha256=1mA_fPmaGabqrNesTDrP4wKzCB9tXx6sf5sHFxIXR-4,9648
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_util.so,sha256=EkZFnBM1iywkrlkovcC8TeVM4yrhZi5kOTJDFR7NBbM,58256
jax_tpu_embedding/sparsecore/lib/core/pybind_input_preprocessing.so,sha256=wPtQC8MeB7xRN_FRkZz9mi3o61f73ynBw5vr6wkfuZQ,11575584
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=uBQfVMj_21cbTbMGgu7N0Z26DzVtxHkIuxr5G6gQD88,4483
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_activation_unstack.py,sha256=ZYd1Adm8YpZvT5u2fhKsn7jMJfb5IvmincnsTQnx5Hw,3004
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_csr.py,sha256=I2NVAoFc_RwAND168Ty4Q0q2h90JuA3x6C9Q8RAB8FI,6027
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad.py,sha256=3xJJAP-9nqLtw5WtBeywJSLjvGSRlIq7km7BptvuyLo,8861
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad_momentum.py,sha256=1eNzGe3NMatv702uatZZjkHe5dkkqQrn5k8hLI1rbdw,12177
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adam.py,sha256=aFPJ9J4fO201_Dx0i5dULq20jEcZYKFITBnpiVayp4I,13589
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_ftrl.py,sha256=U7Ui4UZE6_82IH77zbdrXkK837dDgAJUb1m80gMT1l4,12621
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_laprop.py,sha256=yGU9PQRiYQplUWmAJI1VPuoP4emUgGeZ9SdVzVY4-y4,13524
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_sgd.py,sha256=O0bLsyQIrPnH_dF5DMgmTg9CWvrvkWzMuuxDXyT22AI,7916
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_gradient_stack.py,sha256=rVhI8SSuhdRIP0_gHyWAluYO2CeqGQTgnJ9IL_M6Zuk,2665
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_optimizer_grad.py,sha256=yoIF9k1EvOiUKS_iZgjETPezO6K7zci6bfdDaNJ4VRA,8732
jax_tpu_embedding/sparsecore/lib/core/primitives/utils.py,sha256=0dGgk4k19PEdgdmI7M3OmSLMQOf89N8BB091xvCZWxo,5180
jax_tpu_embedding/sparsecore/lib/fdo/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/fdo/fdo_client.py,sha256=R2zHjXCh4DZE43ppi8lZX1vjPpNQEpLHLYLS7T371Y8,3057
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/linen/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/flax/linen/embed.py,sha256=pO0ST7wOMljlbHmfcHm6ynluKzh94FFJAf5wYR4WnnY,8888
jax_tpu_embedding/sparsecore/lib/flax/linen/embed_optimizer.py,sha256=GmMTx29ZizmNZ9bKUJ92biqNFKnPpydoP8Noge-7L6o,3048
jax_tpu_embedding/sparsecore/lib/flax/nnx/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/flax/nnx/embed.py,sha256=EmTzTw1h6yPhz44_pIxGUoDRuEVEDg7rceW6pNgV3Ac,9191
jax_tpu_embedding/sparsecore/lib/nn/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/nn/embedding.py,sha256=EdzKkfuAC4DX-wvDgXNTmGYIzchmSYgONIVUyceLdug,61626
jax_tpu_embedding/sparsecore/lib/nn/embedding_spec.py,sha256=lqjLLW6lowkJfE7hsUcvue06OzBvS0bo0uNO_2vsi1c,31760
jax_tpu_embedding/sparsecore/lib/nn/table_stacking.py,sha256=S0cGoF4aKT3SI_-tZlwFnCTdFIVvdaT06gi0GafMWiA,43137
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=fMl2WSoE-Q4rbHNXxXxYLbWpowLflRRZEii1ia2Ohbg,3232
jax_tpu_embedding/sparsecore/utils/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/utils/utils.py,sha256=KY71inRnWcep4H8lzZbr6x2egJp54vwzK1wzCv9GK98,2428
jax_tpu_embedding-0.1.0.dev20260312.dist-info/METADATA,sha256=C2o041Nuh_VaBaFG8K7zYuFcA9yJgjWgQzGBRr-2SqY,1284
jax_tpu_embedding-0.1.0.dev20260312.dist-info/WHEEL,sha256=d8qfxAYNremL8SQSnHvaL8xVTVQoVPrd5l-B5VHk1co,135
jax_tpu_embedding-0.1.0.dev20260312.dist-info/top_level.txt,sha256=YzWDDKrd9_GgdRp8EplkirWEw34u_Z7Fxffd1Fdb4OE,18
jax_tpu_embedding-0.1.0.dev20260312.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
jax_tpu_embedding-0.1.0.dev20260312.dist-info/RECORD,,
