CMakeLists.txt
MANIFEST.in
ReadMe.md
pyproject.toml
setup.py
./merlin_sok.egg-info/PKG-INFO
./merlin_sok.egg-info/SOURCES.txt
./merlin_sok.egg-info/dependency_links.txt
./merlin_sok.egg-info/requires.txt
./merlin_sok.egg-info/top_level.txt
./sparse_operation_kit/__init__.py
./sparse_operation_kit/kit_lib.py
./sparse_operation_kit/core/__init__.py
./sparse_operation_kit/core/_version.py
./sparse_operation_kit/core/context_scope.py
./sparse_operation_kit/core/embedding_layer_handle.py
./sparse_operation_kit/core/embedding_variable_v1.py
./sparse_operation_kit/core/embedding_variable_v2.py
./sparse_operation_kit/core/graph_keys.py
./sparse_operation_kit/core/initialize.py
./sparse_operation_kit/core/inplace_initializer.py
./sparse_operation_kit/embeddings/__init__.py
./sparse_operation_kit/embeddings/all2all_dense_embedding.py
./sparse_operation_kit/embeddings/distributed_embedding.py
./sparse_operation_kit/embeddings/embedding_ops.py
./sparse_operation_kit/embeddings/get_embedding_op.py
./sparse_operation_kit/embeddings/tf_distributed_embedding.py
./sparse_operation_kit/operations/__init__.py
./sparse_operation_kit/operations/compat_ops_lib.py
./sparse_operation_kit/optimizers/__init__.py
./sparse_operation_kit/optimizers/adam.py
./sparse_operation_kit/optimizers/base_optimizer.py
./sparse_operation_kit/optimizers/optimizer.py
./sparse_operation_kit/optimizers/utils.py
./sparse_operation_kit/saver/Saver.py
./sparse_operation_kit/saver/__init__.py
./sparse_operation_kit/tf/__init__.py
./sparse_operation_kit/tf/keras/__init__.py
./sparse_operation_kit/tf/keras/mixed_precision/__init__.py
./sparse_operation_kit/tf/keras/mixed_precision/loss_scale_optimizer.py
./sparse_operation_kit/tf/keras/optimizers/__init__.py
./sparse_operation_kit/tf/keras/optimizers/adam.py
./sparse_operation_kit/tf/keras/optimizers/common.py
./sparse_operation_kit/tf/keras/optimizers/lazy_adam.py
cmakes/FindNCCL.cmake
cmakes/FindNVTX.cmake
cmakes/FindTensorFlow.cmake
kit_cc/framework/compat/kernels/gpu_device_functions.h
kit_cc/framework/compat/kernels/gpu_kernel_helper.h
kit_cc/framework/compat/kernels/unique.cu
kit_cc/framework/compat/kernels/unsorted_segment_sum.cu
kit_cc/framework/compat/ops/unique.cc
kit_cc/framework/compat/ops/unsorted_segment_sum.cc
kit_cc/framework/kernels/assign_embedding_variable_op.cc
kit_cc/framework/kernels/bprop.cc
kit_cc/framework/kernels/create_embedding_dense.cc
kit_cc/framework/kernels/create_embedding_sparse.cc
kit_cc/framework/kernels/create_global_adam_optimizer.cc
kit_cc/framework/kernels/create_var.cc
kit_cc/framework/kernels/custom_optimizer_apply_gradients.cc
kit_cc/framework/kernels/dense_fprop.cc
kit_cc/framework/kernels/dump_to_file.cc
kit_cc/framework/kernels/embedding_variable.cc
kit_cc/framework/kernels/embedding_variable.h
kit_cc/framework/kernels/embedding_variable_assign_sub.cu
kit_cc/framework/kernels/gen_random_seed.cc
kit_cc/framework/kernels/get_nccl_unique_id.cc
kit_cc/framework/kernels/load_embedding_values.cc
kit_cc/framework/kernels/optimizer_init.cc
kit_cc/framework/kernels/plugin_init.cc
kit_cc/framework/kernels/read_embedding_variable_op.cc
kit_cc/framework/kernels/restore_from_file.cc
kit_cc/framework/kernels/sparse_fprop.cc
kit_cc/framework/kernels/test_op.cu
kit_cc/framework/ops/assign_embedding_variable_op.cc
kit_cc/framework/ops/bprop.cc
kit_cc/framework/ops/create_embedding_dense.cc
kit_cc/framework/ops/create_embedding_sparse.cc
kit_cc/framework/ops/create_global_adam_optimizer.cc
kit_cc/framework/ops/create_var.cc
kit_cc/framework/ops/custom_optimizer_apply_gradients.cc
kit_cc/framework/ops/dense_fprop.cc
kit_cc/framework/ops/dump_to_file.cc
kit_cc/framework/ops/embedding_variable_assign_sub.cc
kit_cc/framework/ops/init.cc
kit_cc/framework/ops/load_embedding_values.cc
kit_cc/framework/ops/optimizer_init.cc
kit_cc/framework/ops/read_embedding_variable_op.cc
kit_cc/framework/ops/restore_from_file.cc
kit_cc/framework/ops/sparse_fprop.cc
kit_cc/framework/ops/test_op.cc
kit_cc/kit_cc_infra/include/common.cuh
kit_cc/kit_cc_infra/include/common.h
kit_cc/kit_cc_infra/include/facade.h
kit_cc/kit_cc_infra/include/dispatcher/dispatcher_builder.h
kit_cc/kit_cc_infra/include/embeddings/embedding_layer.h
kit_cc/kit_cc_infra/include/embeddings/embedding_lookuper.h
kit_cc/kit_cc_infra/include/embeddings/embedding_lookuper_builder.h
kit_cc/kit_cc_infra/include/embeddings/manager.h
kit_cc/kit_cc_infra/include/hashtable/hashtable.h
kit_cc/kit_cc_infra/include/hashtable/identity_hashtable.h
kit_cc/kit_cc_infra/include/hashtable/nv_hashtable.h
kit_cc/kit_cc_infra/include/hashtable/nv_hashtable.hpp
kit_cc/kit_cc_infra/include/hashtable/simple_hashtable.h
kit_cc/kit_cc_infra/include/hashtable/cudf/LICENSE
kit_cc/kit_cc_infra/include/hashtable/cudf/concurrent_unordered_map.cuh
kit_cc/kit_cc_infra/include/hashtable/cudf/hash_functions.cuh
kit_cc/kit_cc_infra/include/hashtable/cudf/managed.cuh
kit_cc/kit_cc_infra/include/hashtable/cudf/managed_allocator.cuh
kit_cc/kit_cc_infra/include/initializer/constant_initializer.h
kit_cc/kit_cc_infra/include/initializer/initializer_interface.h
kit_cc/kit_cc_infra/include/initializer/random_uniform.h
kit_cc/kit_cc_infra/include/operation/builder_container.h
kit_cc/kit_cc_infra/include/operation/construction_context.h
kit_cc/kit_cc_infra/include/operation/op_context.h
kit_cc/kit_cc_infra/include/operation/operation.h
kit_cc/kit_cc_infra/include/operation/operation_builder.h
kit_cc/kit_cc_infra/include/operation/operation_helper.h
kit_cc/kit_cc_infra/include/operation/operation_interface.h
kit_cc/kit_cc_infra/include/optimizer/adam_optimizer.h
kit_cc/kit_cc_infra/include/optimizer/grad_update_preparer.h
kit_cc/kit_cc_infra/include/optimizer/optimizer_interface.h
kit_cc/kit_cc_infra/include/optimizer/prepare_functions.h
kit_cc/kit_cc_infra/include/optimizer/update_functions.h
kit_cc/kit_cc_infra/include/optimizer/update_preparer.h
kit_cc/kit_cc_infra/include/parameters/manager_interface.h
kit_cc/kit_cc_infra/include/parameters/param_interface.h
kit_cc/kit_cc_infra/include/parameters/raw_manager.h
kit_cc/kit_cc_infra/include/parameters/raw_param.h
kit_cc/kit_cc_infra/include/parameters/raw_state.h
kit_cc/kit_cc_infra/include/parameters/state_interface.h
kit_cc/kit_cc_infra/include/resources/cpu_resource.h
kit_cc/kit_cc_infra/include/resources/event.h
kit_cc/kit_cc_infra/include/resources/event_manager.h
kit_cc/kit_cc_infra/include/resources/gpu_resource.h
kit_cc/kit_cc_infra/include/resources/manager.h
kit_cc/kit_cc_infra/include/resources/mpi_context.h
kit_cc/kit_cc_infra/include/tensor_buffer/embedding_buffer.h
kit_cc/kit_cc_infra/include/tensor_buffer/general_buffer2.hpp
kit_cc/kit_cc_infra/include/tensor_buffer/tensor2.hpp
kit_cc/kit_cc_infra/include/tensor_buffer/tensor2_wrapper.h
kit_cc/kit_cc_infra/include/tensor_buffer/tensor_interface.h
kit_cc/kit_cc_infra/include/tensor_buffer/tf_tensor_wrapper.h
kit_cc/kit_cc_infra/src/common.cc
kit_cc/kit_cc_infra/src/common.cu
kit_cc/kit_cc_infra/src/facade.cc
kit_cc/kit_cc_infra/src/embeddings/embedding_layer.cc
kit_cc/kit_cc_infra/src/embeddings/embedding_lookuper.cc
kit_cc/kit_cc_infra/src/embeddings/manager.cc
kit_cc/kit_cc_infra/src/hashtable/identity_hashtable.cu
kit_cc/kit_cc_infra/src/hashtable/nv_hashtable.cc
kit_cc/kit_cc_infra/src/hashtable/nv_hashtable.cu
kit_cc/kit_cc_infra/src/hashtable/simple_hashtable.cu
kit_cc/kit_cc_infra/src/initializer/constant_initializer.cu
kit_cc/kit_cc_infra/src/initializer/initializer_interface.cc
kit_cc/kit_cc_infra/src/initializer/random_uniform.cu
kit_cc/kit_cc_infra/src/operation/builder_container.cc
kit_cc/kit_cc_infra/src/operation/construction_context.cc
kit_cc/kit_cc_infra/src/operation/op_context.cc
kit_cc/kit_cc_infra/src/operation/operation.cc
kit_cc/kit_cc_infra/src/operation/operation_helper.cc
kit_cc/kit_cc_infra/src/optimizer/adam_optimizer.cc
kit_cc/kit_cc_infra/src/optimizer/grad_update_preparer.cc
kit_cc/kit_cc_infra/src/optimizer/optimizer_interface.cc
kit_cc/kit_cc_infra/src/optimizer/prepare_functions.cu
kit_cc/kit_cc_infra/src/optimizer/update_functions.cu
kit_cc/kit_cc_infra/src/parameters/manager_interface.cc
kit_cc/kit_cc_infra/src/parameters/param_interface.cc
kit_cc/kit_cc_infra/src/parameters/raw_manager.cc
kit_cc/kit_cc_infra/src/parameters/raw_param.cc
kit_cc/kit_cc_infra/src/parameters/raw_state.cc
kit_cc/kit_cc_infra/src/resources/cpu_resource.cc
kit_cc/kit_cc_infra/src/resources/event.cc
kit_cc/kit_cc_infra/src/resources/event_manager.cc
kit_cc/kit_cc_infra/src/resources/gpu_resource.cc
kit_cc/kit_cc_infra/src/resources/manager.cc
kit_cc/kit_cc_infra/src/resources/mpi_context.cc
kit_cc/kit_cc_infra/src/tensor_buffer/embedding_buffer.cc
kit_cc/kit_cc_infra/src/tensor_buffer/tf_tensor_wrapper.cc
kit_cc/kit_cc_infra/third_party/eigen3/unsupported/Eigen/CXX11/src/ThreadPool/SimpleThreadPool.h
kit_cc/kit_cc_infra/third_party/eigen3/unsupported/Eigen/CXX11/src/ThreadPool/ThreadEnvironment.h
kit_cc/kit_cc_infra/third_party/eigen3/unsupported/Eigen/CXX11/src/ThreadPool/ThreadPoolInterface.h
kit_cc_impl/embedding/common/include/backward_functions.cuh
kit_cc_impl/embedding/common/include/backward_functions.h
kit_cc_impl/embedding/common/include/conversion_kernels.cuh
kit_cc_impl/embedding/common/include/dumping_functions.h
kit_cc_impl/embedding/common/include/forward_functions.cuh
kit_cc_impl/embedding/common/include/forward_functions.h
kit_cc_impl/embedding/common/src/backward_functions.cu
kit_cc_impl/embedding/common/src/dumping_functions.cc
kit_cc_impl/embedding/common/src/dumping_functions.cu
kit_cc_impl/embedding/common/src/forward_functions.cu
kit_cc_impl/embedding/dispatcher/all2all_input_dispatcher.cu
kit_cc_impl/embedding/dispatcher/all2all_output_dispatcher.cu
kit_cc_impl/embedding/dispatcher/all_gather_dispatcher.cu
kit_cc_impl/embedding/dispatcher/reduce_scatter_dispatcher.cu
kit_cc_impl/embedding/lookuper/dense_gather.cu
kit_cc_impl/embedding/lookuper/distributed.cc
kit_cc_impl/embedding/operation/csr_conversion_distributed.cu
kit_cc_impl/operations/CMakeLists.txt