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=nxYeb_7EimyIdXPJ5szu3MNNuJ3VYQPX6ImDflUVH6Y,364
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=uqfVolxmqnD7GRdm4G95YX2fW9kFQthAv7_IP5QtnIA,12587
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_core.so,sha256=XR03gGnAfQHImkLCvLp-G4ePNwOYlztCte8LnMzV3Zw,614080
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=CJK9NgKLP0YssAmJbIwa4DQzi9XzLf1-boM0wxXqeik,58328
jax_tpu_embedding/sparsecore/lib/core/pybind_input_preprocessing.so,sha256=46x_4sOSaqdYu15kO4nJcexz9eIluwjFOeQ-ccyWGtg,11623352
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=UoNhiYWM4zXfa1CGRBl_Dj0xcHGB5wd-sud6_WLG5dk,3180
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_csr.py,sha256=xXLIt-m2iqoIbOJXu4ZbuQF04zh6hoB68Z6lHMOSdxg,6353
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad.py,sha256=37ecCSjj-NO4inlWZkmRTU8jChLsAN8UplNplrAJNqc,8956
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad_momentum.py,sha256=Bxuo0RUjN3mXYB4FMkR4v2Ii2J1blu5EyjWMFApHKws,12221
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adam.py,sha256=0iHHkK_MDO1qwE2Jw09bdeOTpIh12Q2tc9jY8uC2Wh4,13822
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_f2a.py,sha256=qmBSEq8dVrzlp9POEO-vV3SaVudAvPK91yd4sjzANp4,12986
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_ftrl.py,sha256=1Y7GwVs2jUbgS0FfhyAeVvRGZ6vCDzzxxSGdzX6aapE,12739
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_laprop.py,sha256=_pNAZWqehc8f_SBOCPBFHfkpsRD04XK_VtwMbsqKpv0,13612
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_sgd.py,sha256=VsnwauRbLB-GEezcQ6TzfYH8OzHtC9z2BeCoqgJElYE,7996
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_gradient_stack.py,sha256=BiMbHTaKPwTqgq7d_fpE0Fy40Fg8HF6_2eMWsh8xxfw,2831
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_optimizer_grad.py,sha256=_yZZg2X43f_xA_Fy35Jo4tclyp3OeaJTR45kUWV2siM,9931
jax_tpu_embedding/sparsecore/lib/core/primitives/utils.py,sha256=U1g-S7-ErmbW7HK1BPRL8Ff6qPgklqYxDYGrwaeoMGk,5145
jax_tpu_embedding/sparsecore/lib/fdo/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/fdo/csv_file_fdo_client.py,sha256=bLTT1KNkZvqaIxCVQfM3nJb1A6JT10QiuZwZov5KPFA,7667
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=hzx1Kf1ZGaRKe2tdO4mN63C0-azZ1Z-JkXLIX8s_cJI,9630
jax_tpu_embedding/sparsecore/lib/nn/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/nn/embedding.py,sha256=7Mr0fOsCZpGu93QJ1aMla833oNIXAG0wOwfPwEsbXOI,69777
jax_tpu_embedding/sparsecore/lib/nn/embedding_spec.py,sha256=xYEt0kKfoXRwieTtqjjLoY8iQdQx_h9-6W0SIcl4HiQ,36269
jax_tpu_embedding/sparsecore/lib/nn/table_stacking.py,sha256=dfXYDDWZpsTGuEjOvh4XE13GYZKdcGKPtoTs3YoQ3W4,44316
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=ziZQlk8raIv0dqkd3PgqPGtqVgmDTeHGkP67XWYCa40,4015
jax_tpu_embedding-0.1.0.dist-info/METADATA,sha256=GLqKerJ8rni7puJY8q3WUnVXaOXuqDOmZaTozGYN2ls,3374
jax_tpu_embedding-0.1.0.dist-info/WHEEL,sha256=HQMRSg_k05ezWXPH1XUJ_laKhoKpSlhfLEBLEo0Ob5I,135
jax_tpu_embedding-0.1.0.dist-info/top_level.txt,sha256=YzWDDKrd9_GgdRp8EplkirWEw34u_Z7Fxffd1Fdb4OE,18
jax_tpu_embedding-0.1.0.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
jax_tpu_embedding-0.1.0.dist-info/RECORD,,
