All modules for which code is available

• sparse_operation_kit.core.context_scope
• sparse_operation_kit.core.initialize
• sparse_operation_kit.embeddings.all2all_dense_embedding
• sparse_operation_kit.embeddings.distributed_embedding
• sparse_operation_kit.optimizers.utils
• sparse_operation_kit.saver.Saver