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=QgyHybnEr9c0KHgoaxaoK9eXQJlAkCHSFqGtN1hRcWw,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=uA7lkdFTTSSTCT-HUF1pCpfEVvN3KCe0SzBnLIc9FoE,9221
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=M-gjeaw_OJgwZkjw3EupVux1Ycbt53wBKJ6B_Ih9x9E,4185
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=K30_LgbEVsZjv7QT0Z7iC6YUWDgU6m43mt_roEBzEVY,14024
jax_tpu_embedding/sparsecore/lib/core/libinput_preprocessing_core.so,sha256=DEk1mbrI3a3G6Jn1NbkCCeld6lBAZNg7ZuzhW6ooxEM,613496
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=vfJrwXeF0qDMyveLzObsa2kMOScjQXyjYeBhcG16j6Q,60888
jax_tpu_embedding/sparsecore/lib/core/pybind_input_preprocessing.so,sha256=LNcHS52_QwCZJVI8ckHhzm7sQ_EiTuws63TxZYQMIaU,11623792
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=KiNumuLAN7759OOXRXxjBsaZZxeFu6fKrS_23WEnzV4,3230
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_csr.py,sha256=Ju3wMYZZtg5OuGQPu1r3SCsQmApRxoUwXTsOkLgMOVo,6653
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad.py,sha256=kTKjssqxXeum7eJ41znJPqXPU4L9YWXGsr8IrWcT-fA,8560
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adagrad_momentum.py,sha256=QohCscIdUnEdXCdDrNA2nuzDdEHOBrtaioXh61oJfa8,11372
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_adam.py,sha256=MortdDe8hJDt9_iQOV6s5YvXUoR2LCxj-Nz906fH-HM,12002
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_f2a.py,sha256=4QIOFpEywSYviFGSQermeHZ5XEcs2Xe6ZAsfyeR6MMI,12172
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_ftrl.py,sha256=ipjq7s1KPcOqG2zq3H2AW2YI_1ou6fAT1YTi0MM7giM,11803
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_laprop.py,sha256=Q8rXAOGsya7HbzgGsZh5P-sZ02WUHhu4cJ3ZgDtFlpk,11792
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_grad_with_sgd.py,sha256=Um1Uy_gFqfTYHwFQHGeSW8Gmuz4aFZTF5gTW8j-W-y4,6835
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_gradient_stack.py,sha256=a70kiL_ncHmeOY2VyiayyqJW_qRU2QB5Egf0Wak-zyg,2908
jax_tpu_embedding/sparsecore/lib/core/primitives/sparse_dense_matmul_optimizer_grad.py,sha256=XEMPN1ZGLePJaj1ZLyosUbIptmh3VjTvMTOGA8W27js,9406
jax_tpu_embedding/sparsecore/lib/core/primitives/utils.py,sha256=e4u44lC8ogLYBh0uU5MkGhG--Hy0griVNH7MpY8T0VU,6866
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=fwbiJYO-tatuo7_EMDoMPiXYL1kwlKfIsXpsflptwM4,7892
jax_tpu_embedding/sparsecore/lib/fdo/fdo_client.py,sha256=R2zHjXCh4DZE43ppi8lZX1vjPpNQEpLHLYLS7T371Y8,3057
jax_tpu_embedding/sparsecore/lib/fdo/file_fdo_client.py,sha256=FkQ9wywhoeepr67YP31xHAqvqt5fn9KOdYHvWn18EeY,8202
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=o9017haibU3Oxem7-2SkInK-rARyfLPm3xx6ScbYEVs,14238
jax_tpu_embedding/sparsecore/lib/flax/linen/embed_optimizer.py,sha256=afa-chnYmEMX-ZazeJBe62Xo4LweVUxLu1jffOpWHLw,3057
jax_tpu_embedding/sparsecore/lib/flax/nnx/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/flax/nnx/checkpoint_utils.py,sha256=bXAItpel9-LzDa6b8dMwB5f0qU9YHoG5PjmCeIJU9zc,40753
jax_tpu_embedding/sparsecore/lib/flax/nnx/embed.py,sha256=dpCnqv6ctHlwXqXk5V-wUViitRVbDFywTQjqvnlJPXI,9630
jax_tpu_embedding/sparsecore/lib/flax/nnx/offline_converter.py,sha256=Ms_09tisnpd85fyEaBlTI0b7sijCCIBwiviBkROsOH0,4645
jax_tpu_embedding/sparsecore/lib/nn/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/lib/nn/custom_optimizer_lowering.py,sha256=3vbZCdM7FMjx3tgHPh87zDjmBckQh6l8XUQ5h2P-Zws,5495
jax_tpu_embedding/sparsecore/lib/nn/embedding.py,sha256=h8k2PVj5f_uvxmoQNziSFDr7L_lCFkXlIFzm6xm0ows,69818
jax_tpu_embedding/sparsecore/lib/nn/embedding_proto_utils.py,sha256=1kG2F5CblAm3WIKWNrltADTIut0-ShRfuYNT9fIRnZs,10140
jax_tpu_embedding/sparsecore/lib/nn/embedding_spec.py,sha256=FvPxBVSb56ldg8ZFJ6pqOYTa7N7y7_Ch6_JmfD9ahgo,43746
jax_tpu_embedding/sparsecore/lib/nn/table_stacking.py,sha256=tgfDzuWHMPfhvEa6UPJjJuC7yv_OsUPHZ4sixIYeQN4,45824
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=0WK10KlC-nfRDnmfjq8E9U59Y110CnmRMghTDtY5998,5614
jax_tpu_embedding/sparsecore/utils/__init__.py,sha256=UzqtnquXbj4s_gh8To-RuuiPxpt-BPEVP3YHbss5pcE,663
jax_tpu_embedding/sparsecore/utils/utils.py,sha256=M5k-3gByATB_utb8nPfs3mlffbbbHhKTq3OPw104gmU,7837
jax_tpu_embedding-0.2.0.dist-info/METADATA,sha256=-vQZ7s4DzFhBPt-6ABS5UL72NoaiorMOHCJ7fHHWOss,3316
jax_tpu_embedding-0.2.0.dist-info/WHEEL,sha256=jGQN0dd4ddELyRs7-TbsiO4c3ojUVZ99hgFoXn8CVNk,113
jax_tpu_embedding-0.2.0.dist-info/top_level.txt,sha256=YzWDDKrd9_GgdRp8EplkirWEw34u_Z7Fxffd1Fdb4OE,18
jax_tpu_embedding-0.2.0.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
jax_tpu_embedding-0.2.0.dist-info/RECORD,,
