├── .bazelrc ├── .bazelversion ├── .clang-format ├── .devcontainer ├── Dockerfile └── devcontainer.json ├── .dockerignore ├── .github ├── ISSUE_TEMPLATE │ ├── bug-performance-report.md │ └── feature-request.md ├── boring-cyborg.yml ├── pull_request_template.md ├── release-drafter.yml └── workflows │ ├── backport.yml │ ├── check-bazel-format.yml │ ├── check-clang-format.yml │ ├── check-python-format.yml │ ├── ci_test.yml │ ├── release-drafter.yml │ └── release.yml ├── .gitignore ├── .vscode └── extensions.json ├── BUILD ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── STYLE_GUIDE.md ├── VERSION.txt ├── WORKSPACE ├── configure ├── configure.py ├── deepray ├── BUILD ├── __init__.py ├── activations │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── hardshrink.py │ ├── lisht.py │ ├── mish.py │ ├── rrelu.py │ ├── snake.py │ ├── softshrink.py │ ├── sparsemax.py │ ├── swish.py │ ├── tanhshrink.py │ └── tests │ │ ├── __init__.py │ │ ├── activations_test.py │ │ ├── hardshrink_test.py │ │ ├── lisht_test.py │ │ ├── mish_test.py │ │ ├── rrelu_test.py │ │ ├── run_all_test.py │ │ ├── snake_test.py │ │ ├── softshrink_test.py │ │ ├── sparsemax_test.py │ │ ├── swish_test.py │ │ └── tanhshrink_test.py ├── callbacks │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── average_model_checkpoint.py │ ├── callbacks.py │ ├── model_checkpoint.py │ ├── profiler_callback.py │ ├── progbar_logger.py │ ├── tests │ │ ├── __init__.py │ │ ├── avg_model_checkpoint_test.py │ │ ├── run_all_test.py │ │ ├── time_stopping_test.py │ │ └── tqdm_progress_bar_test.py │ ├── time_history.py │ ├── time_stopping.py │ ├── tqdm_progress_bar.py │ └── training_speed.py ├── conftest.py ├── copts.bzl ├── core │ ├── BUILD │ ├── __init__.py │ ├── common │ │ ├── __init__.py │ │ ├── distribution_utils.py │ │ ├── distribution_utils_test.py │ │ ├── flags.py │ │ ├── registry_imports.py │ │ └── streamz_counters.py │ ├── compile_utils.py │ ├── export_saved_model.py │ ├── export_saved_model_test.py │ ├── modeling_tf_utils.py │ ├── platform │ │ ├── BUILD │ │ ├── build_config.default.bzl │ │ ├── build_config_root.bzl │ │ ├── build_config_root.default.bzl │ │ └── rules_cc.bzl │ ├── runner.py │ ├── standard_runner.py │ ├── standard_runner_test.py │ ├── trainer.py │ └── utils │ │ ├── __init__.py │ │ ├── clip.py │ │ ├── common.py │ │ ├── common_test.py │ │ ├── epoch_helper.py │ │ ├── loop_fns.py │ │ ├── misc │ │ ├── __init__.py │ │ ├── callstack_sampler.py │ │ ├── distribution_utils.py │ │ ├── distribution_utils_test.py │ │ ├── model_helpers.py │ │ ├── model_helpers_test.py │ │ └── tpu_lib.py │ │ ├── summary_manager.py │ │ ├── summary_manager_interface.py │ │ ├── tpu_summaries.py │ │ └── tpu_summaries_test.py ├── custom_ops │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── correlation_cost │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── correlation_cost_op.cc │ │ │ │ ├── correlation_cost_op.h │ │ │ │ └── correlation_cost_op_gpu.cu.cc │ │ │ └── ops │ │ │ │ └── correlation_cost_op.cc │ │ └── python │ │ │ ├── __init__.py │ │ │ ├── optical_flow.py │ │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── optical_flow_test.py │ │ │ └── run_all_test.py │ ├── embedding_bag │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── embedding_bag_backward_kernels.cu.cc │ │ │ │ ├── embedding_bag_ops.cc │ │ │ │ ├── embedding_bag_ops.h │ │ │ │ └── embedding_bag_ops_gpu.cu.cc │ │ │ └── ops │ │ │ │ └── embedding_bag_ops.cc │ │ └── python │ │ │ ├── __init__.py │ │ │ ├── embedding_bag.py │ │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── embedding_bag_test.py │ │ │ └── run_all_test.py │ ├── embedding_variable │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── embedding │ │ │ │ ├── BUILD │ │ │ │ ├── batch.cu.cc │ │ │ │ ├── batch.h │ │ │ │ ├── bloom_filter_policy.h │ │ │ │ ├── cache.h │ │ │ │ ├── cache_factory.h │ │ │ │ ├── cache_thread_pool_creator.h │ │ │ │ ├── config.proto │ │ │ │ ├── counter_filter_descriptor_impl.h │ │ │ │ ├── counter_filter_policy.h │ │ │ │ ├── cpu_hash_map_kv.h │ │ │ │ ├── dense_hash_map_kv.h │ │ │ │ ├── dram_leveldb_storage.h │ │ │ │ ├── dram_pmem_storage.h │ │ │ │ ├── dram_ssd_storage.h │ │ │ │ ├── dynamic_dim_feature_descriptor_impl.h │ │ │ │ ├── emb_file.h │ │ │ │ ├── emb_file_creator.h │ │ │ │ ├── embedding_config.h │ │ │ │ ├── embedding_memory_pool.h │ │ │ │ ├── embedding_var.cu.cc │ │ │ │ ├── embedding_var.h │ │ │ │ ├── embedding_var_ckpt_data.cc │ │ │ │ ├── embedding_var_ckpt_data.h │ │ │ │ ├── embedding_var_context.h │ │ │ │ ├── embedding_var_dump_iterator.h │ │ │ │ ├── embedding_var_restore.cc │ │ │ │ ├── embedding_var_restore.h │ │ │ │ ├── eviction_manager.h │ │ │ │ ├── feature_descriptor.h │ │ │ │ ├── feature_descriptor_impl.h │ │ │ │ ├── filter_factory.h │ │ │ │ ├── filter_policy.h │ │ │ │ ├── globalstep_shrink_policy.h │ │ │ │ ├── gpu_hash_map_kv.h │ │ │ │ ├── gpu_hash_table.cu.cc │ │ │ │ ├── gpu_hash_table.h │ │ │ │ ├── hbm_dram_ssd_storage.h │ │ │ │ ├── hbm_dram_storage.h │ │ │ │ ├── hbm_multi_tier_feature_descriptor.h │ │ │ │ ├── hbm_storage_iterator.h │ │ │ │ ├── intra_thread_copy_id_allocator.h │ │ │ │ ├── kv_interface.h │ │ │ │ ├── l2weight_shrink_policy.h │ │ │ │ ├── leveldb_kv.h │ │ │ │ ├── multi_tier_storage.cu.cc │ │ │ │ ├── multi_tier_storage.h │ │ │ │ ├── normal_feature_descriptor.h │ │ │ │ ├── nullable_filter_policy.h │ │ │ │ ├── shrink_policy.h │ │ │ │ ├── single_tier_storage.h │ │ │ │ ├── ssd_hash_kv.h │ │ │ │ ├── ssd_record_descriptor.cc │ │ │ │ ├── ssd_record_descriptor.h │ │ │ │ ├── storage.h │ │ │ │ ├── storage_config.h │ │ │ │ └── storage_factory.h │ │ │ ├── fused_embedding │ │ │ │ ├── embedding_lookup_sparse_local_op.cc │ │ │ │ ├── embedding_lookup_sparse_local_op_test.cc │ │ │ │ ├── embedding_lookup_sparse_op.h │ │ │ │ ├── embedding_lookup_sparse_post_grad_op_test.cc │ │ │ │ ├── embedding_lookup_sparse_post_op.cc │ │ │ │ ├── embedding_lookup_sparse_post_op_test.cc │ │ │ │ ├── embedding_lookup_sparse_pre_op.cc │ │ │ │ ├── embedding_lookup_sparse_pre_op_test.cc │ │ │ │ ├── fused_embedding_common.cu.h │ │ │ │ ├── fused_embedding_local_ops_gpu.cu.cc │ │ │ │ ├── fused_embedding_local_ops_test.cc │ │ │ │ ├── fused_embedding_ops.cc │ │ │ │ ├── fused_embedding_post_grad_ops_test.cc │ │ │ │ ├── fused_embedding_post_ops_gpus.cu.cc │ │ │ │ ├── fused_embedding_post_ops_test.cc │ │ │ │ ├── fused_embedding_pre_ops_gpus.cu.cc │ │ │ │ └── fused_embedding_pre_ops_test.cc │ │ │ ├── fused_layer_norm │ │ │ │ ├── BUILD │ │ │ │ ├── compile_util.h │ │ │ │ ├── fused_layer_normalize_ops.cc │ │ │ │ └── fused_layer_normalize_ops_test.cc │ │ │ ├── group_embedding │ │ │ │ ├── group_embedding_lookup_ops.cc │ │ │ │ ├── group_embedding_lookup_ops.cu.cc │ │ │ │ ├── group_embedding_lookup_ops_test.cc │ │ │ │ ├── group_embedding_lookup_sparse_backward_base_ops.cu.h │ │ │ │ ├── group_embedding_lookup_sparse_backward_ops.cc │ │ │ │ ├── group_embedding_lookup_sparse_backward_ops.cu.cc │ │ │ │ ├── group_embedding_lookup_sparse_forward_base_ops.cu.h │ │ │ │ ├── group_embedding_lookup_sparse_forward_base_ops.h │ │ │ │ ├── group_embedding_lookup_sparse_forward_ops.cc │ │ │ │ └── group_embedding_lookup_sparse_forward_ops.cu.cc │ │ │ ├── incr_save_restore │ │ │ │ ├── incr_save_restore_ops.cc │ │ │ │ ├── incr_save_restore_ops.h │ │ │ │ └── incr_save_restore_ops_test.cc │ │ │ ├── kernels │ │ │ │ ├── embedding_collection.cc │ │ │ │ ├── hotness_calculate.cu.cc │ │ │ │ ├── hotness_calculate.h │ │ │ │ ├── kv_variable_lookup_ops.cc │ │ │ │ ├── kv_variable_ops.cc │ │ │ │ ├── kv_variable_restore_ops.cc │ │ │ │ ├── kv_variable_util.cc │ │ │ │ ├── kv_variable_util.h │ │ │ │ ├── save_restore_ops.cc │ │ │ │ ├── save_restore_tensor_ev.h │ │ │ │ ├── training_adagrad_ops.cc │ │ │ │ ├── training_adam_async_ops.cc │ │ │ │ ├── training_adam_ops.cc │ │ │ │ ├── training_ali_op_helpers.h │ │ │ │ ├── training_ali_ops_gpu.cu.cc │ │ │ │ ├── training_ali_ops_gpu.h │ │ │ │ ├── training_ftrl_ops.cc │ │ │ │ └── training_sgd_ops.cc │ │ │ ├── lib │ │ │ │ ├── BUILD │ │ │ │ ├── allocator.cc │ │ │ │ ├── allocator.h │ │ │ │ ├── buffered_file.h │ │ │ │ ├── ev_allocator.cc │ │ │ │ ├── ev_allocator.h │ │ │ │ ├── ev_allocator_gpu.cu.cc │ │ │ │ ├── ev_allocator_test.cc │ │ │ │ ├── experimental_pmem_allocator.cc │ │ │ │ ├── experimental_pmem_allocator.h │ │ │ │ ├── experimental_pmem_allocator_utils.cc │ │ │ │ ├── experimental_pmem_allocator_utils.h │ │ │ │ ├── pmem_allocator.cc │ │ │ │ ├── pmem_allocator.h │ │ │ │ ├── tensor_bundle.cc │ │ │ │ ├── tensor_bundle.h │ │ │ │ └── tensor_bundle_test.cc │ │ │ ├── ops │ │ │ │ ├── embedding_collection.cc │ │ │ │ ├── group_embedding_ops.cc │ │ │ │ ├── incr_save_restore_ops.cc │ │ │ │ ├── kv_variable_ops.cc │ │ │ │ ├── save_restore_ops.cc │ │ │ │ ├── training_adagrad_ops.cc │ │ │ │ ├── training_adam_async_ops.cc │ │ │ │ ├── training_adam_ops.cc │ │ │ │ ├── training_ftrl_ops.cc │ │ │ │ └── training_sgd_ops.cc │ │ │ └── tests │ │ │ │ ├── BUILD │ │ │ │ ├── embedding_variable_memory_test.cc │ │ │ │ ├── embedding_variable_ops_test.cc │ │ │ │ ├── embedding_variable_performance_test.cc │ │ │ │ └── embedding_variable_test.h │ │ ├── config.proto │ │ ├── embedding_variable_ops_test.py │ │ ├── multiplex_1_test.py │ │ ├── python │ │ │ ├── __init__.py │ │ │ ├── group_embedding_lookup_ops.py │ │ │ ├── group_embedding_types.py │ │ │ ├── kv_variable_ops.py │ │ │ └── tests │ │ │ │ ├── __init__.py │ │ │ │ ├── embedding_bag_test.py │ │ │ │ ├── group_embedding_lookup_ops_test.py │ │ │ │ └── run_all_test.py │ │ ├── variable_scope.py │ │ └── variables.py │ ├── ffm_ops │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── ffm_kernels.cc │ │ │ │ ├── ffm_kernels.cu.cc │ │ │ │ └── ffm_kernels.h │ │ │ └── ops │ │ │ │ └── ffm_ops.cc │ │ └── python │ │ │ ├── __init__.py │ │ │ ├── ffm_ops.py │ │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── feature_cross_test.py │ │ │ ├── ffm_ops_test.py │ │ │ └── run_all_test.py │ ├── multiplex_1 │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── multiplex_1_kernel.cc │ │ ├── multiplex_1_op.cc │ │ ├── multiplex_1_op.py │ │ └── multiplex_1_test.py │ ├── multiplex_2 │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── multiplex_2_kernel.cc │ │ ├── multiplex_2_kernel.cu.cc │ │ ├── multiplex_2_kernel.h │ │ ├── multiplex_2_op.cc │ │ ├── multiplex_2_op.py │ │ └── multiplex_2_test.py │ ├── multiplex_3 │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── multiplex_3_kernel.cc │ │ ├── multiplex_3_op.cc │ │ ├── multiplex_3_op.py │ │ └── multiplex_3_test.py │ ├── multiplex_4 │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── model_using_multiplex.py │ │ ├── multiplex_2_save.py │ │ ├── multiplex_4_kernel.cc │ │ ├── multiplex_4_load_use.py │ │ ├── multiplex_4_op.cc │ │ ├── multiplex_4_op.py │ │ └── multiplex_4_test.py │ ├── parquet_dataset │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── arrow_util.cc │ │ │ │ ├── arrow_util.h │ │ │ │ ├── eigen.h │ │ │ │ ├── parquet_batch_reader.cc │ │ │ │ ├── parquet_batch_reader.h │ │ │ │ ├── parquet_dataset_ops.cc │ │ │ │ ├── parquet_dataset_ops.h │ │ │ │ └── parquet_pybind.cc │ │ │ └── ops │ │ │ │ └── parquet_ops.cc │ │ ├── python │ │ │ ├── __init__.py │ │ │ ├── dataframe.py │ │ │ ├── parquet_dataset_ops.py │ │ │ ├── parquet_pybind.py │ │ │ └── tests │ │ │ │ ├── parquet_dataset_ops_test.py │ │ │ │ └── run_all_test.py │ │ └── read_parquet_deepray.py │ ├── seq2seq │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── beam_search_ops.cc │ │ │ │ ├── beam_search_ops.h │ │ │ │ └── beam_search_ops_gpu.cu.cc │ │ │ └── ops │ │ │ │ └── beam_search_ops.cc │ │ └── python │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── attention_wrapper.py │ │ │ ├── basic_decoder.py │ │ │ ├── beam_search_decoder.py │ │ │ ├── decoder.py │ │ │ ├── loss.py │ │ │ ├── sampler.py │ │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── attention_wrapper_test.py │ │ │ ├── basic_decoder_test.py │ │ │ ├── beam_search_decoder_test.py │ │ │ ├── beam_search_ops_test.py │ │ │ ├── decoder_test.py │ │ │ ├── loss_test.py │ │ │ └── run_all_test.py │ ├── simple_hash_table │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── simple_hash_table.py │ │ ├── simple_hash_table_kernel.cc │ │ ├── simple_hash_table_op.cc │ │ ├── simple_hash_table_op.py │ │ └── simple_hash_table_test.py │ ├── sleep │ │ ├── BUILD │ │ ├── README.md │ │ ├── sleep_bin.py │ │ ├── sleep_kernel.cc │ │ ├── sleep_op.cc │ │ ├── sleep_op.py │ │ └── sleep_test.py │ ├── text │ │ ├── BUILD │ │ └── cc │ │ │ ├── kernels │ │ │ ├── parse_time_kernel.cc │ │ │ └── skip_gram_kernels.cc │ │ │ └── ops │ │ │ ├── parse_time_op.cc │ │ │ └── skip_gram_ops.cc │ ├── training_ops │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── training_ops.cc │ │ │ │ ├── training_ops.h │ │ │ │ └── training_ops_gpu.cu.cc │ │ │ └── ops │ │ │ │ └── training_ops.cc │ │ └── python │ │ │ ├── __init__.py │ │ │ └── training_ops.py │ ├── unique_ops │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ │ ├── kernels │ │ │ │ ├── task_runner.h │ │ │ │ ├── unique_ali_op.cc │ │ │ │ ├── unique_ali_op_gpu.cu.cc │ │ │ │ └── unique_ali_op_util.h │ │ │ └── ops │ │ │ │ └── unique_ops.cc │ │ └── python │ │ │ ├── __init__.py │ │ │ ├── tests │ │ │ ├── __init__.py │ │ │ ├── run_all_test.py │ │ │ └── unique_ali_op_test.py │ │ │ └── unique_ops.py │ ├── utils │ │ ├── BUILD │ │ ├── check.h │ │ ├── fake_input.cc │ │ ├── fake_input.h │ │ ├── kernel_benchmark_testlib.cc │ │ ├── kernel_benchmark_testlib.h │ │ ├── ok_status_util.h │ │ ├── ops_testutil.cc │ │ ├── ops_testutil.h │ │ ├── ops_testutil_test.cc │ │ ├── random.cc │ │ ├── random.h │ │ ├── spin_lock.h │ │ ├── spin_rw_lock.h │ │ ├── tensor_testutil.cc │ │ ├── tensor_testutil.h │ │ └── tensor_testutil_test.cc │ └── zero_out │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── cc │ │ ├── kernels │ │ │ └── zero_out_kernels.cc │ │ └── ops │ │ │ └── zero_out_ops.cc │ │ └── python │ │ ├── __init__.py │ │ ├── ops │ │ ├── __init__.py │ │ └── zero_out_ops.py │ │ └── tests │ │ ├── __init__.py │ │ ├── run_all_test.py │ │ └── zero_out_ops_test.py ├── datasets │ ├── BUILD │ ├── __init__.py │ ├── adult_census_income │ │ ├── __init__.py │ │ ├── adult_census_income.py │ │ ├── adult_census_income_test.py │ │ └── feature_map.csv │ ├── ali-ccp │ │ ├── README.md │ │ ├── __init__.py │ │ ├── ali_ccp.py │ │ ├── ali_ccp_test.py │ │ └── preprocess_ali_ccp.py │ ├── ali_display_ad_click │ │ ├── README.md │ │ ├── __init__.py │ │ ├── ali_display_ad_click.py │ │ ├── ali_display_ad_click_test.py │ │ ├── feature_map.csv │ │ └── processing │ │ │ ├── prep_1_backbone.py │ │ │ ├── prep_2_bahavior.py │ │ │ ├── prep_3_merge.py │ │ │ ├── prep_4_sort.py │ │ │ ├── run.sh │ │ │ ├── stats.py │ │ │ └── to_tfrecord.py │ ├── amazon_books_2014 │ │ ├── README.md │ │ ├── __init__.py │ │ ├── amazon_books_2014.py │ │ ├── amazon_books_2014_test.py │ │ ├── defaults.py │ │ ├── download_amazon_books_2014.sh │ │ ├── feature_map.csv │ │ └── preprocessing │ │ │ ├── __init__.py │ │ │ ├── io.py │ │ │ ├── ops.py │ │ │ ├── parquet_to_tfrecord.py │ │ │ └── sim_preprocessing.py │ ├── avazu │ │ ├── README.md │ │ ├── __init__.py │ │ ├── avazu.py │ │ ├── avazu_test.py │ │ ├── feature_map.csv │ │ └── processing.py │ ├── bookscorpus │ │ └── __init__.py │ ├── cifar │ │ ├── __init__.py │ │ ├── cifar.py │ │ └── cifar_test.py │ ├── creditcardfraud │ │ ├── __init__.py │ │ ├── creditcardfraud.py │ │ └── creditcardfraud_test.py │ ├── criteo │ │ ├── Dockerfile_preprocessing │ │ ├── __init__.py │ │ ├── criteo.py │ │ ├── criteo_dataset.md │ │ ├── criteo_test.py │ │ ├── criteo_tsv_reader.py │ │ ├── criteo_tsv_reader_test.py │ │ ├── feature_map.csv │ │ ├── feature_map_small.csv │ │ ├── feature_map_xlarge.csv │ │ ├── preproc │ │ │ ├── DGX-2_config.sh │ │ │ ├── DGX-A100_config.sh │ │ │ ├── NVT_shuffle_spark.py │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── defaults.py │ │ │ │ └── feature_spec.py │ │ │ ├── gpu │ │ │ │ ├── get_gpu_resources.sh │ │ │ │ └── spark-defaults.conf │ │ │ ├── parquet_to_binary.py │ │ │ ├── prepare_dataset.sh │ │ │ ├── preproc_NVTabular.py │ │ │ ├── run_NVTabular.sh │ │ │ ├── run_spark.sh │ │ │ ├── run_spark_cpu.sh │ │ │ ├── run_spark_gpu_DGX-2.sh │ │ │ ├── run_spark_gpu_DGX-A100.sh │ │ │ ├── spark_data_utils.py │ │ │ ├── split_dataset.py │ │ │ └── verify_criteo_downloaded.sh │ │ ├── requirements_preprocessing.txt │ │ └── run.sh │ ├── csv_pipeline │ │ ├── __init__.py │ │ └── csv_pipeline.py │ ├── datapipeline.py │ ├── downloader │ │ ├── Downloader.py │ │ ├── GooglePretrainedWeightDownloader.py │ │ ├── NVIDIAPretrainedWeightDownloader.py │ │ ├── TextSharding.py │ │ ├── bertPrep.py │ │ ├── bookscorpus │ │ │ ├── BooksDownloader.py │ │ │ ├── BookscorpusTextFormatting.py │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── download_files.py │ │ │ ├── download_list.py │ │ │ ├── epub2txt.py │ │ │ ├── make_sentlines.py │ │ │ ├── requirements.txt │ │ │ ├── tokenize_sentlines.py │ │ │ └── url_list.jsonl │ │ ├── classifier_data_lib.py │ │ ├── create_datasets_from_start.sh │ │ ├── create_finetuning_data.py │ │ ├── create_pretraining_data.py │ │ ├── download.sh │ │ ├── glue │ │ │ ├── GLUEDownloader.py │ │ │ └── __init__.py │ │ ├── process.sh │ │ ├── pubmed │ │ │ ├── PubMedDownloader.py │ │ │ ├── PubMedTextFormatting.py │ │ │ └── __init__.py │ │ ├── squad │ │ │ ├── SquadDownloader.py │ │ │ └── __init__.py │ │ ├── tokenization.py │ │ └── wikicorpus │ │ │ ├── WikiDownloader.py │ │ │ ├── WikicorpusTextFormatting.py │ │ │ └── __init__.py │ ├── enwik8 │ │ └── __init__.py │ ├── fashion_mnist │ │ ├── __init__.py │ │ ├── fashion_mnist.py │ │ └── fashion_mnist_test.py │ ├── glue │ │ └── __init__.py │ ├── imagenet-1k │ │ ├── __init__.py │ │ └── imagenet_to_gcs.py │ ├── imdb │ │ ├── __init__.py │ │ ├── imdb.py │ │ └── imdb_test.py │ ├── kafka_pipeline │ │ ├── __init__.py │ │ ├── kafka_pipeline.py │ │ └── kafka_pipeline_test.py │ ├── mnist │ │ ├── __init__.py │ │ ├── mnist.py │ │ └── mnist_test.py │ ├── movielens │ │ ├── __init__.py │ │ ├── constants.py │ │ ├── data_pipeline.py │ │ ├── data_preprocessing.py │ │ ├── movielens.csv │ │ ├── movielens.py │ │ ├── movielens_100k_ratings.py │ │ ├── movielens_1m_ratings.py │ │ ├── movielens_ratings_test.py │ │ ├── popen_helper.py │ │ ├── process.py │ │ ├── producer.py │ │ └── stat_utils.py │ ├── openwebtext │ │ ├── README.md │ │ ├── __init__.py │ │ ├── build_openwebtext_pretraining_dataset.py │ │ ├── build_pretraining_dataset.py │ │ ├── openwebtext.py │ │ ├── openwebtext_test.py │ │ ├── run.sh │ │ ├── tokenization.py │ │ ├── util │ │ │ ├── __init__.py │ │ │ ├── training_utils.py │ │ │ └── utils.py │ │ └── vocab.txt │ ├── parquet_pipeline │ │ ├── __init__.py │ │ ├── ali_parquet_dataset.py │ │ ├── ali_parquet_dataset_test.py │ │ └── parquet_pipeline_test.py │ ├── pubmed │ │ └── __init__.py │ ├── squad │ │ ├── __init__.py │ │ ├── classifier_dataset.py │ │ ├── pretrain_dataset.py │ │ ├── squad.csv │ │ ├── squad.py │ │ ├── squad_dataset.py │ │ ├── squad_lib.py │ │ ├── squad_lib_sp.py │ │ ├── squad_test.py │ │ ├── v1.1 │ │ │ ├── evaluate-v1.1.py │ │ │ └── squad_v1.1_meta_data │ │ └── v2.0 │ │ │ ├── evaluate-v2.0.py │ │ │ └── squad_v2.0_meta_data │ ├── taobao │ │ ├── README.md │ │ ├── __init__.py │ │ ├── csv_price_int32_to_parquet.py │ │ └── csv_price_string_to_parquet.py │ ├── tfrecord_pipeline │ │ ├── __init__.py │ │ ├── tfrecord_pipeline.py │ │ └── tfrecord_pipeline_test.py │ ├── tokenization.py │ ├── toxic_comment_classification_challenge │ │ ├── __init__.py │ │ ├── toxic_comment_classification_challenge.py │ │ └── toxic_comment_classification_challenge_test.py │ ├── wikicorpus_en │ │ ├── README.md │ │ ├── processing │ │ │ ├── BookscorpusTextFormatting.py │ │ │ ├── Downloader.py │ │ │ ├── TextSharding.py │ │ │ ├── WikiDownloader.py │ │ │ ├── WikicorpusTextFormatting.py │ │ │ ├── __init__.py │ │ │ ├── build_pretraining_dataset.py │ │ │ ├── create_datasets_from_start.sh │ │ │ ├── dataPrep.py │ │ │ ├── file_utils.py │ │ │ ├── run.sh │ │ │ ├── tokenization.py │ │ │ ├── tokenization_utils.py │ │ │ ├── utils.py │ │ │ ├── vocab │ │ │ │ └── vocab.txt │ │ │ └── wikiextractor │ │ │ │ ├── .gitignore │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── WikiExtractor.py │ │ │ │ ├── categories.filter │ │ │ │ ├── cirrus-extract.py │ │ │ │ └── extract.sh │ │ ├── wikicorpus_en.py │ │ └── wikicorpus_en_test.py │ ├── wikitext103 │ │ ├── __init__.py │ │ └── wikitext103.py │ └── wikitext2 │ │ └── __init__.py ├── deepray.bzl ├── design_patterns.py ├── layers │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── adaptive_pooling.py │ ├── attention.py │ ├── bucketize.py │ ├── compositional_embedding.py │ ├── crf.py │ ├── dcn.py │ ├── dense.py │ ├── dense_einsum.py │ ├── dot_interaction.py │ ├── dynamic_embedding.py │ ├── embedding.py │ ├── embedding_variable.py │ ├── esn.py │ ├── feature_cross.py │ ├── flash_attention.py │ ├── masked_softmax.py │ ├── max_unpooling_2d.py │ ├── max_unpooling_2d_v2.py │ ├── maxout.py │ ├── mlp.py │ ├── netvlad.py │ ├── networks │ │ ├── __init__.py │ │ ├── albert_encoder.py │ │ ├── albert_encoder_test.py │ │ ├── bert_dense_encoder_test.py │ │ ├── bert_encoder.py │ │ ├── bert_encoder_test.py │ │ ├── fnet.py │ │ ├── fnet_test.py │ │ ├── funnel_transformer.py │ │ ├── funnel_transformer_test.py │ │ ├── mobile_bert_encoder.py │ │ ├── mobile_bert_encoder_test.py │ │ ├── packed_sequence_embedding.py │ │ ├── packed_sequence_embedding_test.py │ │ ├── sparse_mixer.py │ │ ├── sparse_mixer_test.py │ │ ├── xlnet_base.py │ │ └── xlnet_base_test.py │ ├── noisy_dense.py │ ├── normalizations.py │ ├── on_device_embedding.py │ ├── poincare.py │ ├── polynomial.py │ ├── pooling.py │ ├── position_embedding.py │ ├── relative_attention.py │ ├── rnn │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── esn_cell.py │ │ ├── layer_norm_lstm_cell.py │ │ ├── layer_norm_simple_rnn_cell.py │ │ ├── nas_cell.py │ │ ├── peephole_lstm_cell.py │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── esn_cell_test.py │ │ │ ├── layer_norm_lstm_cell_test.py │ │ │ ├── layer_norm_simple_rnn_cell_test.py │ │ │ ├── nas_cell_test.py │ │ │ ├── peephole_lstm_cell_test.py │ │ │ └── run_all_test.py │ ├── self_attention_mask.py │ ├── snake.py │ ├── sparsemax.py │ ├── spatial_pyramid_pooling.py │ ├── spectral_normalization.py │ ├── stochastic_depth.py │ ├── tests_bak │ │ ├── __init__.py │ │ ├── adaptive_pooling_test.py │ │ ├── attention_test.py │ │ ├── crf_test.py │ │ ├── dcn_test.py │ │ ├── dense_einsum_test.py │ │ ├── dnn_test.py │ │ ├── dot_interaction_test.py │ │ ├── embedding_test.py │ │ ├── esn_test.py │ │ ├── feature_cross_test.py │ │ ├── masked_softmax_test.py │ │ ├── max_unpooling_2d_test.py │ │ ├── max_unpooling_2d_v2_test.py │ │ ├── maxout_test.py │ │ ├── mlp_test.py │ │ ├── netvlad_test.py │ │ ├── noisy_dense_test.py │ │ ├── normalizations_test.py │ │ ├── on_device_embedding_test.py │ │ ├── poincare_test.py │ │ ├── polynomial_test.py │ │ ├── position_embedding_test.py │ │ ├── relative_attention_test.py │ │ ├── run_all_test.py │ │ ├── snake_test.py │ │ ├── sparsemax_test.py │ │ ├── spatial_pyramid_pooling_test.py │ │ ├── spectral_normalization_test.py │ │ ├── stochastic_depth_test.py │ │ ├── tlu_test.py │ │ ├── transformer_scaffold_test.py │ │ ├── transformer_test.py │ │ ├── transformer_xl_test.py │ │ └── wrappers_test.py │ ├── tf_utils.py │ ├── tlu.py │ ├── transformer.py │ ├── transformer_scaffold.py │ ├── transformer_xl.py │ └── wrappers.py ├── losses │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── _loss_util.py │ ├── contrastive.py │ ├── focal_loss.py │ ├── giou_loss.py │ ├── kappa_loss.py │ ├── lifted.py │ ├── losses_impl.py │ ├── metric_learning.py │ ├── npairs.py │ ├── quantiles.py │ ├── softmax_loss.py │ ├── sparsemax_loss.py │ ├── tests │ │ ├── __init__.py │ │ ├── contrastive_test.py │ │ ├── focal_loss_test.py │ │ ├── giou_loss_test.py │ │ ├── kappa_loss_test.py │ │ ├── lifted_test.py │ │ ├── metric_test.py │ │ ├── npairs_test.py │ │ ├── quantiles_test.py │ │ ├── run_all_test.py │ │ ├── sparsemax_loss_test.py │ │ ├── triplet_test.py │ │ └── weighted_sparse_categorical_crossentropy_test.py │ ├── triplet.py │ ├── utils.py │ └── weighted_sparse_categorical_crossentropy.py ├── metrics │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── _ranking.py │ ├── alpha_dcg.py │ ├── arp.py │ ├── cohens_kappa.py │ ├── dcg.py │ ├── f_scores.py │ ├── geometric_mean.py │ ├── hamming.py │ ├── harmonic_mean.py │ ├── hits.py │ ├── matthews_correlation_coefficient.py │ ├── mean_average_precision.py │ ├── metrics_impl.py │ ├── mrr.py │ ├── multilabel_confusion_matrix.py │ ├── ndcg.py │ ├── opa.py │ ├── precision.py │ ├── precision_ia.py │ ├── r_square.py │ ├── recall.py │ ├── streaming_correlations.py │ ├── tests │ │ ├── __init__.py │ │ ├── cohens_kappa_test.py │ │ ├── f_scores_test.py │ │ ├── geometric_mean_test.py │ │ ├── hamming_test.py │ │ ├── harmonic_mean_test.py │ │ ├── matthews_correlation_coefficient_test.py │ │ ├── metrics_test.py │ │ ├── multilabel_confusion_matrix_test.py │ │ ├── r_square_test.py │ │ ├── run_all_test.py │ │ └── streaming_correlations_test.py │ └── utils.py ├── models │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── albert_transformer_encoder.py │ ├── bert_classifier.py │ ├── bert_pretrainer.py │ ├── bert_span_labeler.py │ ├── classification.py │ ├── deep_cross_net.py │ ├── encoder_scaffold.py │ ├── framework.py │ ├── generative │ │ ├── __init__.py │ │ └── conditional_gan.py │ ├── masked_lm.py │ ├── ncf_common.py │ ├── ncf_model.py │ ├── ncf_test.py │ ├── neumf_model.py │ ├── rec │ │ ├── __init__.py │ │ ├── deepfm.py │ │ ├── dien_model.py │ │ ├── din_model.py │ │ ├── dual_channels_deep_model.py │ │ ├── flen.py │ │ ├── flend.py │ │ ├── sequential_recommender_model.py │ │ └── sim_model.py │ ├── span_labeling.py │ ├── stable_diffusion │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __internal__ │ │ │ ├── __init__.py │ │ │ └── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── attention_block.py │ │ │ │ ├── padded_conv2d.py │ │ │ │ └── resnet_block.py │ │ ├── clip_tokenizer.py │ │ ├── constants.py │ │ ├── decoder.py │ │ ├── diffusion_model.py │ │ ├── image_encoder.py │ │ ├── noise_scheduler.py │ │ ├── stable_diffusion.py │ │ ├── stable_diffusion_test.py │ │ └── text_encoder.py │ ├── tests │ │ ├── __init__.py │ │ ├── albert_transformer_encoder_test.py │ │ ├── bert_classifier_test.py │ │ ├── bert_pretrainer_test.py │ │ ├── bert_span_labeler_test.py │ │ ├── classification_test.py │ │ ├── encoder_scaffold_test.py │ │ ├── masked_lm_test.py │ │ ├── span_labeling_test.py │ │ └── transformer_encoder_test.py │ ├── transformer_encoder.py │ └── word2vec.py ├── ops │ ├── __init__.py │ ├── beam_search.py │ ├── beam_search_test.py │ ├── decoding_module.py │ ├── decoding_module_test.py │ ├── sampling_module.py │ ├── segment_extractor.py │ └── segment_extractor_test.py ├── optimizers │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── adabelief.py │ ├── adagrad.py │ ├── adam.py │ ├── adam_async.py │ ├── average_wrapper.py │ ├── cocob.py │ ├── conditional_gradient.py │ ├── constants.py │ ├── cyclical_learning_rate.py │ ├── ev_optimizer_patch.py │ ├── ftrl.py │ ├── gradient_descent.py │ ├── lamb.py │ ├── lazy_adam.py │ ├── lookahead.py │ ├── moving_average.py │ ├── multi_optimizer.py │ ├── novograd.py │ ├── optimization.py │ ├── proximal_adagrad.py │ ├── rectified_adam.py │ ├── stochastic_weight_averaging.py │ ├── tests │ │ ├── __init__.py │ │ ├── adabelief_test.py │ │ ├── adam_test.py │ │ ├── cocob_test.py │ │ ├── conditional_gradient_test.py │ │ ├── cyclical_learning_rate_test.py │ │ ├── lamb_test.py │ │ ├── lazy_adam_test.py │ │ ├── lookahead_test.py │ │ ├── moving_average_test.py │ │ ├── multi_optimizer_test.py │ │ ├── novograd_test.py │ │ ├── proximal_adagrad_test.py │ │ ├── rectified_adam_test.py │ │ ├── run_all_test.py │ │ ├── standard_test.py │ │ ├── stochastic_weight_averaging_test.py │ │ ├── weight_decay_optimizers_test.py │ │ └── yogi_test.py │ ├── utils.py │ ├── warmup.py │ ├── weight_decay_optimizers.py │ └── yogi.py ├── options.py ├── register.py ├── testing │ ├── BUILD │ ├── __init__.py │ ├── serialization.py │ └── tests │ │ ├── __init__.py │ │ ├── run_all_test.py │ │ └── serialization_test.py ├── tests │ ├── __init__.py │ ├── register_test.py │ └── run_all_test.py ├── text │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── crf.py │ ├── crf_wrapper.py │ ├── parse_time_op.py │ ├── skip_gram_ops.py │ └── tests │ │ ├── __init__.py │ │ ├── crf_test.py │ │ ├── crf_wrapper_test.py │ │ ├── parse_time_op_test.py │ │ ├── run_all_test.py │ │ └── skip_gram_ops_test.py ├── utils │ ├── BUILD │ ├── README.md │ ├── __init__.py │ ├── accelerator │ │ ├── __init__.py │ │ ├── tpu.py │ │ └── tpu_test.py │ ├── benchmark.py │ ├── ckpt_util.py │ ├── data │ │ ├── __init__.py │ │ ├── feature_map.py │ │ ├── feature_map_test.py │ │ ├── file_io.py │ │ ├── file_io_test.py │ │ └── input_meta.py │ ├── ensure_tf_install.py │ ├── export │ │ ├── __init__.py │ │ ├── export.py │ │ └── export_test.py │ ├── flags │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _base.py │ │ ├── _benchmark.py │ │ ├── _conventions.py │ │ ├── _data.py │ │ ├── _device.py │ │ ├── _distribution.py │ │ ├── _misc.py │ │ ├── _performance.py │ │ ├── common_flags.py │ │ ├── core.py │ │ ├── flags_test.py │ │ └── guidelines.md │ ├── gpu_affinity.py │ ├── horovod_utils.py │ ├── keras_utils.py │ ├── logging_util.py │ ├── logs │ │ ├── __init__.py │ │ ├── cloud_lib.py │ │ ├── cloud_lib_test.py │ │ ├── guidelines.md │ │ ├── hooks_helper.py │ │ ├── hooks_helper_test.py │ │ ├── logger.py │ │ ├── logger_test.py │ │ ├── mlperf_helper.py │ │ └── summary_manager.py │ ├── misc │ │ ├── __init__.py │ │ ├── model_helpers.py │ │ └── model_helpers_test.py │ ├── resource_loader.py │ ├── test_utils.py │ ├── tests │ │ ├── __init__.py │ │ ├── keras_utils_test.py │ │ ├── run_all_test.py │ │ └── test_utils_test.py │ ├── tfrecord_inspecter.py │ ├── timer.py │ └── types.py ├── version.py ├── workspace0.bzl ├── workspace2.bzl └── workspace3.bzl ├── docker.sh ├── docs ├── README.md ├── overview.md └── tutorials │ ├── README.md │ ├── _template.ipynb │ ├── _toc.yaml │ ├── average_optimizers_callback.ipynb │ ├── image_ops.ipynb │ ├── layers_normalizations.ipynb │ ├── layers_weightnormalization.ipynb │ ├── losses_triplet.ipynb │ ├── networks_seq2seq_nmt.ipynb │ ├── optimizers_conditionalgradient.ipynb │ ├── optimizers_cyclicallearningrate.ipynb │ ├── optimizers_lazyadam.ipynb │ ├── time_stopping.ipynb │ └── tqdm_progress_bar.ipynb ├── modelzoo ├── CV │ ├── Classify_images_of_clothing │ │ ├── run_horovod.sh │ │ └── train.py │ ├── GAN │ │ ├── conditional_gan_mnist.py │ │ └── train.py │ ├── SwinTransformers │ │ ├── model.py │ │ ├── run_horovod.sh │ │ └── train.py │ ├── __init__.py │ ├── knowledge_distillation │ │ └── knowledge_distillation.py │ └── mnist │ │ ├── __init__.py │ │ ├── run_early.sh │ │ ├── run_horovod.sh │ │ └── train.py ├── LanguageModeling │ ├── BERT │ │ ├── .dockerignore │ │ ├── .gitignore │ │ ├── Dockerfile │ │ ├── README.md │ │ ├── bert_dllogger.json │ │ ├── classifier_data_lib.py │ │ ├── common_flags.py │ │ ├── create_finetuning_data.py │ │ ├── create_pretraining_data.py │ │ ├── data │ │ │ ├── BooksDownloader.py │ │ │ ├── BookscorpusTextFormatting.py │ │ │ ├── Downloader.py │ │ │ ├── GLUEDownloader.py │ │ │ ├── GooglePretrainedWeightDownloader.py │ │ │ ├── NVIDIAPretrainedWeightDownloader.py │ │ │ ├── PubMedDownloader.py │ │ │ ├── PubMedTextFormatting.py │ │ │ ├── README.md │ │ │ ├── SquadDownloader.py │ │ │ ├── TextSharding.py │ │ │ ├── WikiDownloader.py │ │ │ ├── WikicorpusTextFormatting.py │ │ │ ├── __init__.py │ │ │ ├── bertPrep.py │ │ │ ├── create_biobert_datasets_from_start.sh │ │ │ ├── create_datasets_from_start.sh │ │ │ └── images │ │ │ │ ├── bert_pipeline.png │ │ │ │ └── images_nvlamb.png │ │ ├── gpu_affinity.py │ │ ├── input_pipeline.py │ │ ├── model_saving_utils.py │ │ ├── official │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── hyperparams │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── params_dict.py │ │ │ │ │ └── params_dict_test.py │ │ │ │ └── training │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── distributed_executor.py │ │ │ └── nlp │ │ │ │ ├── bert_modeling.py │ │ │ │ ├── bert_models.py │ │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ └── losses │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── weighted_sparse_categorical_crossentropy.py │ │ │ │ │ └── weighted_sparse_categorical_crossentropy_test.py │ │ │ │ └── transformer │ │ │ │ ├── __init__.py │ │ │ │ ├── beam_search_v1.py │ │ │ │ ├── beam_search_v1_test.py │ │ │ │ ├── model_params.py │ │ │ │ ├── model_utils.py │ │ │ │ └── model_utils_test.py │ │ ├── optimization.py │ │ ├── run.sub │ │ ├── run_classifier.py │ │ ├── run_pretraining.py │ │ ├── run_squad.py │ │ ├── scripts │ │ │ ├── benchmark_pretraining_lamb_phase2.sh │ │ │ ├── configs │ │ │ │ ├── configurations.yml │ │ │ │ ├── pretrain_config.sh │ │ │ │ └── squad_config.sh │ │ │ ├── data_download.sh │ │ │ ├── docker │ │ │ │ ├── build.sh │ │ │ │ └── launch.sh │ │ │ ├── finetune_inference_benchmark.sh │ │ │ ├── finetune_train_benchmark.sh │ │ │ ├── run_inference_benchmark.sh │ │ │ ├── run_inference_benchmark_seq128.sh │ │ │ ├── run_pretraining_adam.sh │ │ │ ├── run_pretraining_lamb.sh │ │ │ ├── run_pretraining_lamb_phase1.sh │ │ │ ├── run_pretraining_lamb_phase2.sh │ │ │ ├── run_squad.sh │ │ │ └── run_squad_inference.sh │ │ ├── squad_lib.py │ │ ├── squad_lib_sp.py │ │ ├── tf_trt.py │ │ └── tokenization.py │ ├── Classify-text-with-BERT │ │ └── classify_text_with_bert.py │ ├── ELECTRA │ │ ├── .gitignore │ │ ├── Dockerfile │ │ ├── LICENSE │ │ ├── NOTICE │ │ ├── README.md │ │ ├── build_pretraining_dataset.py │ │ ├── configuration.py │ │ ├── configuration_utils.py │ │ ├── data │ │ │ ├── BooksDownloader.py │ │ │ ├── BookscorpusTextFormatting.py │ │ │ ├── Downloader.py │ │ │ ├── GooglePretrainedWeightDownloader.py │ │ │ ├── MRPCDownloader.py │ │ │ ├── NVIDIAPretrainedWeightDownloader.py │ │ │ ├── SquadDownloader.py │ │ │ ├── TextSharding.py │ │ │ ├── WikiDownloader.py │ │ │ ├── WikicorpusTextFormatting.py │ │ │ ├── __init__.py │ │ │ ├── create_datasets_from_start.sh │ │ │ ├── dataPrep.py │ │ │ ├── glue │ │ │ │ └── download_mrpc.sh │ │ │ └── squad │ │ │ │ └── squad_download.sh │ │ ├── file_utils.py │ │ ├── gpu_affinity.py │ │ ├── images │ │ │ └── total_loss.svg │ │ ├── modeling.py │ │ ├── modeling_utils.py │ │ ├── optimization.py │ │ ├── postprocess_pretrained_ckpt.py │ │ ├── pretrain_utils.py │ │ ├── run.sub │ │ ├── run_inference.py │ │ ├── run_pretraining.py │ │ ├── run_tf_squad.py │ │ ├── scripts │ │ │ ├── benchmark_pretraining.sh │ │ │ ├── benchmark_squad.sh │ │ │ ├── bind.sh │ │ │ ├── configs │ │ │ │ ├── pretrain_config.sh │ │ │ │ └── squad_config.sh │ │ │ ├── docker │ │ │ │ ├── build.sh │ │ │ │ └── launch.sh │ │ │ ├── finetune_ckpts_on_squad.sh │ │ │ ├── run_pretraining.sh │ │ │ └── run_squad.sh │ │ ├── squad_utils.py │ │ ├── tokenization.py │ │ ├── tokenization_utils.py │ │ ├── utils.py │ │ └── vocab │ │ │ └── vocab.txt │ ├── Multi-label-classification-with-BERT │ │ ├── README.md │ │ ├── run_horovod.sh │ │ └── trainer.py │ ├── Neural-machine-translation-with-Transformer │ │ ├── a.py │ │ ├── models.py │ │ ├── run.py │ │ └── run_dp.py │ ├── Word2Vec │ │ ├── dataset.py │ │ ├── model.py │ │ ├── run.sh │ │ ├── train.py │ │ └── word2vec.py │ └── text_generation_with_miniature_gpt │ │ └── text_generation_with_miniature_gpt.py ├── Recommendation │ ├── CreditCardFraudDetection │ │ ├── run.sh │ │ └── train.py │ ├── Criteo_DCN │ │ ├── README.md │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── custom_dataset.py │ │ │ └── custom_dataset_test.py │ │ ├── dcn_v2.py │ │ ├── feature_map_small.csv │ │ ├── run.sh │ │ ├── train.py │ │ └── train1.py │ ├── MovieLens │ │ ├── mymodel.py │ │ ├── run.sh │ │ ├── run_ranking.sh │ │ ├── train.py │ │ └── train_ranking.py │ ├── NCF │ │ ├── run_ncf.py │ │ └── run_ncf.sh │ ├── SIM │ │ ├── defaults.py │ │ ├── feature_spec.py │ │ ├── logs │ │ │ ├── tf_training_amazon_books_2014_dien_fp16_gbs32768.230321132836.log │ │ │ ├── tf_training_amazon_books_2014_din_fp16_gbs32768.230321132123.log │ │ │ └── tf_training_amazon_books_2014_sim_fp16_gbs32768.230321133429.log │ │ ├── main.py │ │ ├── run_dien.py │ │ ├── run_din.py │ │ ├── run_horovod.sh │ │ └── run_sim.py │ ├── WideDeep │ │ ├── model.py │ │ ├── run_horovod.sh │ │ └── train.py │ ├── __init__.py │ ├── avazu-ctr-prediction │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── ccpm.py │ │ ├── ccpm_diamond.py │ │ ├── feature_map.csv │ │ ├── run_horovod.sh │ │ └── train.py │ └── ev_test │ │ ├── ev_test.py │ │ └── run.sh ├── __init__.py └── horovod_test │ ├── run.sh │ └── tensorflow2_synthetic_benchmark.py ├── pyproject.toml ├── requirements.txt ├── setup.py ├── third_party ├── BUILD ├── __init__.py ├── arrow │ ├── BUILD │ ├── arrow-20.patch │ └── arrow.BUILD ├── aws-c-common.BUILD ├── aws-c-event-stream.BUILD ├── aws-checksums.BUILD ├── aws-sdk-cpp.BUILD ├── brotli.BUILD ├── clang_toolchain │ ├── BUILD │ ├── cc_configure_clang.bzl │ └── download_clang.bzl ├── cuCollections │ ├── BUILD │ ├── cuCollections.BUILD │ ├── cuco.BUILD │ └── cucollection.patch ├── cutlass.BUILD ├── cxxtimer.BUILD ├── double_conversion.BUILD ├── eigen3.BUILD ├── flash_attn │ ├── BUILD │ ├── flash_attn.BUILD │ └── flash_attn.patch ├── gpus │ ├── BUILD.bazel │ └── find_cuda_config.py ├── hadoop │ ├── BUILD │ ├── LICENSE.txt │ └── hdfs.h ├── leveldb.BUILD ├── libcuckoo.BUILD ├── openblas.BUILD ├── openssl.BUILD ├── oss_c_sdk.BUILD ├── py │ ├── BUILD │ ├── pypi.bzl │ └── python_init_pip.bzl ├── rapidjson.BUILD ├── readerwriterqueue.BUILD ├── remote_config │ ├── BUILD │ └── common.bzl ├── sparsehash_c11 │ ├── BUILD │ ├── sparsehash_c11.BUILD │ └── sparsehash_c11.patch ├── tf │ ├── BUILD │ └── tf_215.patch ├── thrift │ ├── BUILD │ ├── build_defs.bzl │ ├── extra │ │ ├── BUILD │ │ └── config.h │ └── thrift.BUILD ├── xla │ ├── BUILD.bazel │ └── workspace.bzl ├── xsimd.BUILD └── xxHash.BUILD └── tools ├── build_base_container.sh ├── build_dev_container.sh ├── docker ├── base_container.Dockerfile ├── bashrc.bash ├── build_wheel.Dockerfile ├── dev_container.Dockerfile ├── entry.sh └── pre-commit.Dockerfile ├── docs ├── BUILD ├── Readme.md └── build_docs.py ├── format.py ├── install_deps ├── base.packages.txt ├── base.requirements.txt ├── buildifier.sh ├── clang-format.sh ├── devel.packages.txt ├── devel.requirements.txt ├── doc_requirements.txt ├── install_bazelisk.sh ├── install_clang.sh ├── install_cmake.sh ├── install_miniforge.sh ├── install_nsight-systems.sh ├── install_openmpi.sh ├── install_openssh.sh ├── install_patchelf.sh ├── install_python.sh └── setup.packages.sh ├── pre-commit.sh ├── releases ├── horovod_runner_patch.sh └── tf_auditwheel_patch.sh ├── run_build.sh ├── run_google_cloud_tests.sh ├── run_gpu_tests.sh ├── testing └── parallel_gpu_execute.sh └── update_release_version.sh /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.bazelrc -------------------------------------------------------------------------------- /.bazelversion: -------------------------------------------------------------------------------- 1 | 6.5.0 -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.clang-format -------------------------------------------------------------------------------- /.devcontainer/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.devcontainer/Dockerfile -------------------------------------------------------------------------------- /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-performance-report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/ISSUE_TEMPLATE/bug-performance-report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/ISSUE_TEMPLATE/feature-request.md -------------------------------------------------------------------------------- /.github/boring-cyborg.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/boring-cyborg.yml -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/release-drafter.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/release-drafter.yml -------------------------------------------------------------------------------- /.github/workflows/backport.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/backport.yml -------------------------------------------------------------------------------- /.github/workflows/check-bazel-format.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/check-bazel-format.yml -------------------------------------------------------------------------------- /.github/workflows/check-clang-format.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/check-clang-format.yml -------------------------------------------------------------------------------- /.github/workflows/check-python-format.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/check-python-format.yml -------------------------------------------------------------------------------- /.github/workflows/ci_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/ci_test.yml -------------------------------------------------------------------------------- /.github/workflows/release-drafter.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/release-drafter.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/BUILD -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/README.md -------------------------------------------------------------------------------- /STYLE_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/STYLE_GUIDE.md -------------------------------------------------------------------------------- /VERSION.txt: -------------------------------------------------------------------------------- 1 | 0.21.0 -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/WORKSPACE -------------------------------------------------------------------------------- /configure: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/configure -------------------------------------------------------------------------------- /configure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/configure.py -------------------------------------------------------------------------------- /deepray/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/BUILD -------------------------------------------------------------------------------- /deepray/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/__init__.py -------------------------------------------------------------------------------- /deepray/activations/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/BUILD -------------------------------------------------------------------------------- /deepray/activations/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/README.md -------------------------------------------------------------------------------- /deepray/activations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/activations/hardshrink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/hardshrink.py -------------------------------------------------------------------------------- /deepray/activations/lisht.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/lisht.py -------------------------------------------------------------------------------- /deepray/activations/mish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/mish.py -------------------------------------------------------------------------------- /deepray/activations/rrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/rrelu.py -------------------------------------------------------------------------------- /deepray/activations/snake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/snake.py -------------------------------------------------------------------------------- /deepray/activations/softshrink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/softshrink.py -------------------------------------------------------------------------------- /deepray/activations/sparsemax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/sparsemax.py -------------------------------------------------------------------------------- /deepray/activations/swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/swish.py -------------------------------------------------------------------------------- /deepray/activations/tanhshrink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tanhshrink.py -------------------------------------------------------------------------------- /deepray/activations/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/activations/tests/activations_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/activations_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/hardshrink_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/hardshrink_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/lisht_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/lisht_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/mish_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/mish_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/rrelu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/rrelu_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/snake_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/snake_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/softshrink_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/softshrink_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/sparsemax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/sparsemax_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/swish_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/swish_test.py -------------------------------------------------------------------------------- /deepray/activations/tests/tanhshrink_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/activations/tests/tanhshrink_test.py -------------------------------------------------------------------------------- /deepray/callbacks/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/BUILD -------------------------------------------------------------------------------- /deepray/callbacks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/README.md -------------------------------------------------------------------------------- /deepray/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/__init__.py -------------------------------------------------------------------------------- /deepray/callbacks/average_model_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/average_model_checkpoint.py -------------------------------------------------------------------------------- /deepray/callbacks/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/callbacks.py -------------------------------------------------------------------------------- /deepray/callbacks/model_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/model_checkpoint.py -------------------------------------------------------------------------------- /deepray/callbacks/profiler_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/profiler_callback.py -------------------------------------------------------------------------------- /deepray/callbacks/progbar_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/progbar_logger.py -------------------------------------------------------------------------------- /deepray/callbacks/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/callbacks/tests/avg_model_checkpoint_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/tests/avg_model_checkpoint_test.py -------------------------------------------------------------------------------- /deepray/callbacks/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/callbacks/tests/time_stopping_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/tests/time_stopping_test.py -------------------------------------------------------------------------------- /deepray/callbacks/tests/tqdm_progress_bar_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/tests/tqdm_progress_bar_test.py -------------------------------------------------------------------------------- /deepray/callbacks/time_history.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/time_history.py -------------------------------------------------------------------------------- /deepray/callbacks/time_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/time_stopping.py -------------------------------------------------------------------------------- /deepray/callbacks/tqdm_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/tqdm_progress_bar.py -------------------------------------------------------------------------------- /deepray/callbacks/training_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/callbacks/training_speed.py -------------------------------------------------------------------------------- /deepray/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/conftest.py -------------------------------------------------------------------------------- /deepray/copts.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/copts.bzl -------------------------------------------------------------------------------- /deepray/core/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/BUILD -------------------------------------------------------------------------------- /deepray/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/core/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/common/__init__.py -------------------------------------------------------------------------------- /deepray/core/common/distribution_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/common/distribution_utils.py -------------------------------------------------------------------------------- /deepray/core/common/distribution_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/common/distribution_utils_test.py -------------------------------------------------------------------------------- /deepray/core/common/flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/common/flags.py -------------------------------------------------------------------------------- /deepray/core/common/registry_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/common/registry_imports.py -------------------------------------------------------------------------------- /deepray/core/common/streamz_counters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/common/streamz_counters.py -------------------------------------------------------------------------------- /deepray/core/compile_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/compile_utils.py -------------------------------------------------------------------------------- /deepray/core/export_saved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/export_saved_model.py -------------------------------------------------------------------------------- /deepray/core/export_saved_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/export_saved_model_test.py -------------------------------------------------------------------------------- /deepray/core/modeling_tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/modeling_tf_utils.py -------------------------------------------------------------------------------- /deepray/core/platform/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/core/platform/build_config.default.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/platform/build_config.default.bzl -------------------------------------------------------------------------------- /deepray/core/platform/build_config_root.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/platform/build_config_root.bzl -------------------------------------------------------------------------------- /deepray/core/platform/build_config_root.default.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/platform/build_config_root.default.bzl -------------------------------------------------------------------------------- /deepray/core/platform/rules_cc.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/platform/rules_cc.bzl -------------------------------------------------------------------------------- /deepray/core/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/runner.py -------------------------------------------------------------------------------- /deepray/core/standard_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/standard_runner.py -------------------------------------------------------------------------------- /deepray/core/standard_runner_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/standard_runner_test.py -------------------------------------------------------------------------------- /deepray/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/trainer.py -------------------------------------------------------------------------------- /deepray/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/__init__.py -------------------------------------------------------------------------------- /deepray/core/utils/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/clip.py -------------------------------------------------------------------------------- /deepray/core/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/common.py -------------------------------------------------------------------------------- /deepray/core/utils/common_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/common_test.py -------------------------------------------------------------------------------- /deepray/core/utils/epoch_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/epoch_helper.py -------------------------------------------------------------------------------- /deepray/core/utils/loop_fns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/loop_fns.py -------------------------------------------------------------------------------- /deepray/core/utils/misc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/core/utils/misc/callstack_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/misc/callstack_sampler.py -------------------------------------------------------------------------------- /deepray/core/utils/misc/distribution_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/misc/distribution_utils.py -------------------------------------------------------------------------------- /deepray/core/utils/misc/model_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/misc/model_helpers.py -------------------------------------------------------------------------------- /deepray/core/utils/misc/model_helpers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/misc/model_helpers_test.py -------------------------------------------------------------------------------- /deepray/core/utils/misc/tpu_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/misc/tpu_lib.py -------------------------------------------------------------------------------- /deepray/core/utils/summary_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/summary_manager.py -------------------------------------------------------------------------------- /deepray/core/utils/summary_manager_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/summary_manager_interface.py -------------------------------------------------------------------------------- /deepray/core/utils/tpu_summaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/tpu_summaries.py -------------------------------------------------------------------------------- /deepray/core/utils/tpu_summaries_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/core/utils/tpu_summaries_test.py -------------------------------------------------------------------------------- /deepray/custom_ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/correlation_cost/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/correlation_cost/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/correlation_cost/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/correlation_cost/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/correlation_cost/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/correlation_cost/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_bag/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/embedding_bag/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_bag/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/embedding_bag/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_bag/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_bag/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_variable/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/embedding_variable/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_variable/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/embedding_variable/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_variable/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/embedding_variable/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/ffm_ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/ffm_ops/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/ffm_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/ffm_ops/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/ffm_ops/cc/ops/ffm_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/ffm_ops/cc/ops/ffm_ops.cc -------------------------------------------------------------------------------- /deepray/custom_ops/ffm_ops/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/ffm_ops/python/ffm_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/ffm_ops/python/ffm_ops.py -------------------------------------------------------------------------------- /deepray/custom_ops/ffm_ops/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_1/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_1/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_1/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_1/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_1/multiplex_1_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_1/multiplex_1_op.cc -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_1/multiplex_1_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_1/multiplex_1_op.py -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_2/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_2/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_2/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_2/multiplex_2_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_2/multiplex_2_op.cc -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_2/multiplex_2_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_2/multiplex_2_op.py -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_3/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_3/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_3/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_3/multiplex_3_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_3/multiplex_3_op.cc -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_3/multiplex_3_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_3/multiplex_3_op.py -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_4/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_4/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_4/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_4/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_4/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_4/multiplex_4_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_4/multiplex_4_op.cc -------------------------------------------------------------------------------- /deepray/custom_ops/multiplex_4/multiplex_4_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/multiplex_4/multiplex_4_op.py -------------------------------------------------------------------------------- /deepray/custom_ops/parquet_dataset/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/parquet_dataset/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/parquet_dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/parquet_dataset/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/parquet_dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/parquet_dataset/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/parquet_dataset/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/seq2seq/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/seq2seq/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/seq2seq/python/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/python/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/seq2seq/python/decoder.py -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/python/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/seq2seq/python/loss.py -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/python/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/seq2seq/python/sampler.py -------------------------------------------------------------------------------- /deepray/custom_ops/seq2seq/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/simple_hash_table/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/simple_hash_table/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/simple_hash_table/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/simple_hash_table/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/simple_hash_table/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/README.md -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/sleep_bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/sleep_bin.py -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/sleep_kernel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/sleep_kernel.cc -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/sleep_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/sleep_op.cc -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/sleep_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/sleep_op.py -------------------------------------------------------------------------------- /deepray/custom_ops/sleep/sleep_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/sleep/sleep_test.py -------------------------------------------------------------------------------- /deepray/custom_ops/text/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/text/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/text/cc/ops/parse_time_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/text/cc/ops/parse_time_op.cc -------------------------------------------------------------------------------- /deepray/custom_ops/text/cc/ops/skip_gram_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/text/cc/ops/skip_gram_ops.cc -------------------------------------------------------------------------------- /deepray/custom_ops/training_ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/training_ops/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/training_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/training_ops/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/training_ops/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/unique_ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/unique_ops/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/unique_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/unique_ops/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/unique_ops/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/unique_ops/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/utils/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/utils/check.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/check.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/fake_input.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/fake_input.cc -------------------------------------------------------------------------------- /deepray/custom_ops/utils/fake_input.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/fake_input.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/ok_status_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/ok_status_util.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/ops_testutil.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/ops_testutil.cc -------------------------------------------------------------------------------- /deepray/custom_ops/utils/ops_testutil.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/ops_testutil.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/ops_testutil_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/ops_testutil_test.cc -------------------------------------------------------------------------------- /deepray/custom_ops/utils/random.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/random.cc -------------------------------------------------------------------------------- /deepray/custom_ops/utils/random.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/random.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/spin_lock.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/spin_lock.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/spin_rw_lock.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/spin_rw_lock.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/tensor_testutil.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/tensor_testutil.cc -------------------------------------------------------------------------------- /deepray/custom_ops/utils/tensor_testutil.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/tensor_testutil.h -------------------------------------------------------------------------------- /deepray/custom_ops/utils/tensor_testutil_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/utils/tensor_testutil_test.cc -------------------------------------------------------------------------------- /deepray/custom_ops/zero_out/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/zero_out/BUILD -------------------------------------------------------------------------------- /deepray/custom_ops/zero_out/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/custom_ops/zero_out/__init__.py -------------------------------------------------------------------------------- /deepray/custom_ops/zero_out/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/custom_ops/zero_out/python/ops/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /deepray/custom_ops/zero_out/python/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/BUILD -------------------------------------------------------------------------------- /deepray/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/adult_census_income/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/adult_census_income/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/ali-ccp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/ali-ccp/README.md -------------------------------------------------------------------------------- /deepray/datasets/ali-ccp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/ali-ccp/ali_ccp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/ali-ccp/ali_ccp_test.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/ali-ccp/preprocess_ali_ccp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/ali-ccp/preprocess_ali_ccp.py -------------------------------------------------------------------------------- /deepray/datasets/ali_display_ad_click/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/ali_display_ad_click/README.md -------------------------------------------------------------------------------- /deepray/datasets/ali_display_ad_click/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/amazon_books_2014/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/amazon_books_2014/README.md -------------------------------------------------------------------------------- /deepray/datasets/amazon_books_2014/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/amazon_books_2014/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/amazon_books_2014/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/amazon_books_2014/defaults.py -------------------------------------------------------------------------------- /deepray/datasets/amazon_books_2014/preprocessing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/avazu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/avazu/README.md -------------------------------------------------------------------------------- /deepray/datasets/avazu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/avazu/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/avazu/avazu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/avazu/avazu.py -------------------------------------------------------------------------------- /deepray/datasets/avazu/avazu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/avazu/avazu_test.py -------------------------------------------------------------------------------- /deepray/datasets/avazu/feature_map.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/avazu/feature_map.csv -------------------------------------------------------------------------------- /deepray/datasets/avazu/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/avazu/processing.py -------------------------------------------------------------------------------- /deepray/datasets/bookscorpus/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/cifar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/cifar/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/cifar/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/cifar/cifar.py -------------------------------------------------------------------------------- /deepray/datasets/cifar/cifar_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/cifar/cifar_test.py -------------------------------------------------------------------------------- /deepray/datasets/creditcardfraud/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/creditcardfraud/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/Dockerfile_preprocessing: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/Dockerfile_preprocessing -------------------------------------------------------------------------------- /deepray/datasets/criteo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/criteo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/criteo.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/criteo_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/criteo_dataset.md -------------------------------------------------------------------------------- /deepray/datasets/criteo/criteo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/criteo_test.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/criteo_tsv_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/criteo_tsv_reader.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/criteo_tsv_reader_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/criteo_tsv_reader_test.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/feature_map.csv: -------------------------------------------------------------------------------- 1 | name,composition_factor 2 | _c14,"16,16" -------------------------------------------------------------------------------- /deepray/datasets/criteo/feature_map_small.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/feature_map_small.csv -------------------------------------------------------------------------------- /deepray/datasets/criteo/feature_map_xlarge.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/feature_map_xlarge.csv -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/DGX-2_config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/preproc/DGX-2_config.sh -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/data/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/preproc/data/defaults.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/run_NVTabular.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/preproc/run_NVTabular.sh -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/run_spark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/preproc/run_spark.sh -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/run_spark_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/preproc/run_spark_cpu.sh -------------------------------------------------------------------------------- /deepray/datasets/criteo/preproc/split_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/preproc/split_dataset.py -------------------------------------------------------------------------------- /deepray/datasets/criteo/requirements_preprocessing.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | pandas 3 | joblib==0.16 4 | tqdm 5 | -------------------------------------------------------------------------------- /deepray/datasets/criteo/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/criteo/run.sh -------------------------------------------------------------------------------- /deepray/datasets/csv_pipeline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/csv_pipeline/csv_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/csv_pipeline/csv_pipeline.py -------------------------------------------------------------------------------- /deepray/datasets/datapipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/datapipeline.py -------------------------------------------------------------------------------- /deepray/datasets/downloader/Downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/Downloader.py -------------------------------------------------------------------------------- /deepray/datasets/downloader/TextSharding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/TextSharding.py -------------------------------------------------------------------------------- /deepray/datasets/downloader/bertPrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/bertPrep.py -------------------------------------------------------------------------------- /deepray/datasets/downloader/bookscorpus/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/bookscorpus/LICENSE -------------------------------------------------------------------------------- /deepray/datasets/downloader/bookscorpus/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/bookscorpus/README.md -------------------------------------------------------------------------------- /deepray/datasets/downloader/bookscorpus/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/downloader/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/download.sh -------------------------------------------------------------------------------- /deepray/datasets/downloader/glue/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/downloader/process.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/process.sh -------------------------------------------------------------------------------- /deepray/datasets/downloader/pubmed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/downloader/squad/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/downloader/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/downloader/tokenization.py -------------------------------------------------------------------------------- /deepray/datasets/downloader/wikicorpus/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/enwik8/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/fashion_mnist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/fashion_mnist/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/fashion_mnist/fashion_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/fashion_mnist/fashion_mnist.py -------------------------------------------------------------------------------- /deepray/datasets/glue/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/imagenet-1k/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/imagenet-1k/imagenet_to_gcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/imagenet-1k/imagenet_to_gcs.py -------------------------------------------------------------------------------- /deepray/datasets/imdb/__init__.py: -------------------------------------------------------------------------------- 1 | from .imdb import IMDB 2 | -------------------------------------------------------------------------------- /deepray/datasets/imdb/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/imdb/imdb.py -------------------------------------------------------------------------------- /deepray/datasets/imdb/imdb_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/imdb/imdb_test.py -------------------------------------------------------------------------------- /deepray/datasets/kafka_pipeline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/kafka_pipeline/kafka_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/kafka_pipeline/kafka_pipeline.py -------------------------------------------------------------------------------- /deepray/datasets/mnist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/mnist/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/mnist/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/mnist/mnist.py -------------------------------------------------------------------------------- /deepray/datasets/mnist/mnist_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/mnist/mnist_test.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/constants.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/data_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/data_pipeline.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/data_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/data_preprocessing.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/movielens.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/movielens.csv -------------------------------------------------------------------------------- /deepray/datasets/movielens/movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/movielens.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/popen_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/popen_helper.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/process.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/producer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/producer.py -------------------------------------------------------------------------------- /deepray/datasets/movielens/stat_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/movielens/stat_utils.py -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/README.md -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/openwebtext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/openwebtext.py -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/openwebtext_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/openwebtext_test.py -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/run.sh -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/tokenization.py -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/util/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/util/utils.py -------------------------------------------------------------------------------- /deepray/datasets/openwebtext/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/openwebtext/vocab.txt -------------------------------------------------------------------------------- /deepray/datasets/parquet_pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/parquet_pipeline/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/pubmed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/squad/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/squad/classifier_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/classifier_dataset.py -------------------------------------------------------------------------------- /deepray/datasets/squad/pretrain_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/pretrain_dataset.py -------------------------------------------------------------------------------- /deepray/datasets/squad/squad.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/squad.csv -------------------------------------------------------------------------------- /deepray/datasets/squad/squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/squad.py -------------------------------------------------------------------------------- /deepray/datasets/squad/squad_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/squad_dataset.py -------------------------------------------------------------------------------- /deepray/datasets/squad/squad_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/squad_lib.py -------------------------------------------------------------------------------- /deepray/datasets/squad/squad_lib_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/squad_lib_sp.py -------------------------------------------------------------------------------- /deepray/datasets/squad/squad_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/squad_test.py -------------------------------------------------------------------------------- /deepray/datasets/squad/v1.1/evaluate-v1.1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/v1.1/evaluate-v1.1.py -------------------------------------------------------------------------------- /deepray/datasets/squad/v1.1/squad_v1.1_meta_data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/v1.1/squad_v1.1_meta_data -------------------------------------------------------------------------------- /deepray/datasets/squad/v2.0/evaluate-v2.0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/v2.0/evaluate-v2.0.py -------------------------------------------------------------------------------- /deepray/datasets/squad/v2.0/squad_v2.0_meta_data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/squad/v2.0/squad_v2.0_meta_data -------------------------------------------------------------------------------- /deepray/datasets/taobao/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/taobao/README.md -------------------------------------------------------------------------------- /deepray/datasets/taobao/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/tfrecord_pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/tfrecord_pipeline/__init__.py -------------------------------------------------------------------------------- /deepray/datasets/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/tokenization.py -------------------------------------------------------------------------------- /deepray/datasets/wikicorpus_en/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/wikicorpus_en/README.md -------------------------------------------------------------------------------- /deepray/datasets/wikicorpus_en/processing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/wikicorpus_en/processing/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/wikicorpus_en/processing/run.sh -------------------------------------------------------------------------------- /deepray/datasets/wikicorpus_en/wikicorpus_en.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/wikicorpus_en/wikicorpus_en.py -------------------------------------------------------------------------------- /deepray/datasets/wikitext103/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/datasets/wikitext103/wikitext103.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/datasets/wikitext103/wikitext103.py -------------------------------------------------------------------------------- /deepray/datasets/wikitext2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/deepray.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/deepray.bzl -------------------------------------------------------------------------------- /deepray/design_patterns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/design_patterns.py -------------------------------------------------------------------------------- /deepray/layers/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/BUILD -------------------------------------------------------------------------------- /deepray/layers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/README.md -------------------------------------------------------------------------------- /deepray/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/__init__.py -------------------------------------------------------------------------------- /deepray/layers/adaptive_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/adaptive_pooling.py -------------------------------------------------------------------------------- /deepray/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/attention.py -------------------------------------------------------------------------------- /deepray/layers/bucketize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/bucketize.py -------------------------------------------------------------------------------- /deepray/layers/compositional_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/compositional_embedding.py -------------------------------------------------------------------------------- /deepray/layers/crf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/crf.py -------------------------------------------------------------------------------- /deepray/layers/dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/dcn.py -------------------------------------------------------------------------------- /deepray/layers/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/dense.py -------------------------------------------------------------------------------- /deepray/layers/dense_einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/dense_einsum.py -------------------------------------------------------------------------------- /deepray/layers/dot_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/dot_interaction.py -------------------------------------------------------------------------------- /deepray/layers/dynamic_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/dynamic_embedding.py -------------------------------------------------------------------------------- /deepray/layers/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/embedding.py -------------------------------------------------------------------------------- /deepray/layers/embedding_variable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/embedding_variable.py -------------------------------------------------------------------------------- /deepray/layers/esn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/esn.py -------------------------------------------------------------------------------- /deepray/layers/feature_cross.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/feature_cross.py -------------------------------------------------------------------------------- /deepray/layers/flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/flash_attention.py -------------------------------------------------------------------------------- /deepray/layers/masked_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/masked_softmax.py -------------------------------------------------------------------------------- /deepray/layers/max_unpooling_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/max_unpooling_2d.py -------------------------------------------------------------------------------- /deepray/layers/max_unpooling_2d_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/max_unpooling_2d_v2.py -------------------------------------------------------------------------------- /deepray/layers/maxout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/maxout.py -------------------------------------------------------------------------------- /deepray/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/mlp.py -------------------------------------------------------------------------------- /deepray/layers/netvlad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/netvlad.py -------------------------------------------------------------------------------- /deepray/layers/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/layers/networks/albert_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/albert_encoder.py -------------------------------------------------------------------------------- /deepray/layers/networks/albert_encoder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/albert_encoder_test.py -------------------------------------------------------------------------------- /deepray/layers/networks/bert_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/bert_encoder.py -------------------------------------------------------------------------------- /deepray/layers/networks/bert_encoder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/bert_encoder_test.py -------------------------------------------------------------------------------- /deepray/layers/networks/fnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/fnet.py -------------------------------------------------------------------------------- /deepray/layers/networks/fnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/fnet_test.py -------------------------------------------------------------------------------- /deepray/layers/networks/funnel_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/funnel_transformer.py -------------------------------------------------------------------------------- /deepray/layers/networks/mobile_bert_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/mobile_bert_encoder.py -------------------------------------------------------------------------------- /deepray/layers/networks/sparse_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/sparse_mixer.py -------------------------------------------------------------------------------- /deepray/layers/networks/sparse_mixer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/sparse_mixer_test.py -------------------------------------------------------------------------------- /deepray/layers/networks/xlnet_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/xlnet_base.py -------------------------------------------------------------------------------- /deepray/layers/networks/xlnet_base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/networks/xlnet_base_test.py -------------------------------------------------------------------------------- /deepray/layers/noisy_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/noisy_dense.py -------------------------------------------------------------------------------- /deepray/layers/normalizations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/normalizations.py -------------------------------------------------------------------------------- /deepray/layers/on_device_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/on_device_embedding.py -------------------------------------------------------------------------------- /deepray/layers/poincare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/poincare.py -------------------------------------------------------------------------------- /deepray/layers/polynomial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/polynomial.py -------------------------------------------------------------------------------- /deepray/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/pooling.py -------------------------------------------------------------------------------- /deepray/layers/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/position_embedding.py -------------------------------------------------------------------------------- /deepray/layers/relative_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/relative_attention.py -------------------------------------------------------------------------------- /deepray/layers/rnn/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/BUILD -------------------------------------------------------------------------------- /deepray/layers/rnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/README.md -------------------------------------------------------------------------------- /deepray/layers/rnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/__init__.py -------------------------------------------------------------------------------- /deepray/layers/rnn/esn_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/esn_cell.py -------------------------------------------------------------------------------- /deepray/layers/rnn/layer_norm_lstm_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/layer_norm_lstm_cell.py -------------------------------------------------------------------------------- /deepray/layers/rnn/layer_norm_simple_rnn_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/layer_norm_simple_rnn_cell.py -------------------------------------------------------------------------------- /deepray/layers/rnn/nas_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/nas_cell.py -------------------------------------------------------------------------------- /deepray/layers/rnn/peephole_lstm_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/peephole_lstm_cell.py -------------------------------------------------------------------------------- /deepray/layers/rnn/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/layers/rnn/tests/esn_cell_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/tests/esn_cell_test.py -------------------------------------------------------------------------------- /deepray/layers/rnn/tests/nas_cell_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/tests/nas_cell_test.py -------------------------------------------------------------------------------- /deepray/layers/rnn/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/rnn/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/layers/self_attention_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/self_attention_mask.py -------------------------------------------------------------------------------- /deepray/layers/snake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/snake.py -------------------------------------------------------------------------------- /deepray/layers/sparsemax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/sparsemax.py -------------------------------------------------------------------------------- /deepray/layers/spatial_pyramid_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/spatial_pyramid_pooling.py -------------------------------------------------------------------------------- /deepray/layers/spectral_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/spectral_normalization.py -------------------------------------------------------------------------------- /deepray/layers/stochastic_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/stochastic_depth.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/layers/tests_bak/adaptive_pooling_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/adaptive_pooling_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/attention_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/attention_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/crf_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/crf_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/dcn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/dcn_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/dense_einsum_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/dense_einsum_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/dnn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/dnn_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/dot_interaction_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/dot_interaction_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/embedding_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/embedding_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/esn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/esn_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/feature_cross_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/feature_cross_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/masked_softmax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/masked_softmax_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/max_unpooling_2d_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/max_unpooling_2d_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/maxout_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/maxout_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/mlp_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/mlp_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/netvlad_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/netvlad_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/noisy_dense_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/noisy_dense_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/normalizations_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/normalizations_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/poincare_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/poincare_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/polynomial_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/polynomial_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/run_all_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/snake_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/snake_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/sparsemax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/sparsemax_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/stochastic_depth_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/stochastic_depth_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/tlu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/tlu_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/transformer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/transformer_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/transformer_xl_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/transformer_xl_test.py -------------------------------------------------------------------------------- /deepray/layers/tests_bak/wrappers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tests_bak/wrappers_test.py -------------------------------------------------------------------------------- /deepray/layers/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tf_utils.py -------------------------------------------------------------------------------- /deepray/layers/tlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/tlu.py -------------------------------------------------------------------------------- /deepray/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/transformer.py -------------------------------------------------------------------------------- /deepray/layers/transformer_scaffold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/transformer_scaffold.py -------------------------------------------------------------------------------- /deepray/layers/transformer_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/transformer_xl.py -------------------------------------------------------------------------------- /deepray/layers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/layers/wrappers.py -------------------------------------------------------------------------------- /deepray/losses/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/BUILD -------------------------------------------------------------------------------- /deepray/losses/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/README.md -------------------------------------------------------------------------------- /deepray/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/__init__.py -------------------------------------------------------------------------------- /deepray/losses/_loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/_loss_util.py -------------------------------------------------------------------------------- /deepray/losses/contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/contrastive.py -------------------------------------------------------------------------------- /deepray/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/focal_loss.py -------------------------------------------------------------------------------- /deepray/losses/giou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/giou_loss.py -------------------------------------------------------------------------------- /deepray/losses/kappa_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/kappa_loss.py -------------------------------------------------------------------------------- /deepray/losses/lifted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/lifted.py -------------------------------------------------------------------------------- /deepray/losses/losses_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/losses_impl.py -------------------------------------------------------------------------------- /deepray/losses/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/metric_learning.py -------------------------------------------------------------------------------- /deepray/losses/npairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/npairs.py -------------------------------------------------------------------------------- /deepray/losses/quantiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/quantiles.py -------------------------------------------------------------------------------- /deepray/losses/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/softmax_loss.py -------------------------------------------------------------------------------- /deepray/losses/sparsemax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/sparsemax_loss.py -------------------------------------------------------------------------------- /deepray/losses/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/losses/tests/contrastive_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/contrastive_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/focal_loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/focal_loss_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/giou_loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/giou_loss_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/kappa_loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/kappa_loss_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/lifted_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/lifted_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/metric_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/metric_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/npairs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/npairs_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/quantiles_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/quantiles_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/sparsemax_loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/sparsemax_loss_test.py -------------------------------------------------------------------------------- /deepray/losses/tests/triplet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/tests/triplet_test.py -------------------------------------------------------------------------------- /deepray/losses/triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/triplet.py -------------------------------------------------------------------------------- /deepray/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/losses/utils.py -------------------------------------------------------------------------------- /deepray/metrics/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/BUILD -------------------------------------------------------------------------------- /deepray/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/README.md -------------------------------------------------------------------------------- /deepray/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/__init__.py -------------------------------------------------------------------------------- /deepray/metrics/_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/_ranking.py -------------------------------------------------------------------------------- /deepray/metrics/alpha_dcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/alpha_dcg.py -------------------------------------------------------------------------------- /deepray/metrics/arp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/arp.py -------------------------------------------------------------------------------- /deepray/metrics/cohens_kappa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/cohens_kappa.py -------------------------------------------------------------------------------- /deepray/metrics/dcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/dcg.py -------------------------------------------------------------------------------- /deepray/metrics/f_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/f_scores.py -------------------------------------------------------------------------------- /deepray/metrics/geometric_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/geometric_mean.py -------------------------------------------------------------------------------- /deepray/metrics/hamming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/hamming.py -------------------------------------------------------------------------------- /deepray/metrics/harmonic_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/harmonic_mean.py -------------------------------------------------------------------------------- /deepray/metrics/hits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/hits.py -------------------------------------------------------------------------------- /deepray/metrics/mean_average_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/mean_average_precision.py -------------------------------------------------------------------------------- /deepray/metrics/metrics_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/metrics_impl.py -------------------------------------------------------------------------------- /deepray/metrics/mrr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/mrr.py -------------------------------------------------------------------------------- /deepray/metrics/multilabel_confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/multilabel_confusion_matrix.py -------------------------------------------------------------------------------- /deepray/metrics/ndcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/ndcg.py -------------------------------------------------------------------------------- /deepray/metrics/opa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/opa.py -------------------------------------------------------------------------------- /deepray/metrics/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/precision.py -------------------------------------------------------------------------------- /deepray/metrics/precision_ia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/precision_ia.py -------------------------------------------------------------------------------- /deepray/metrics/r_square.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/r_square.py -------------------------------------------------------------------------------- /deepray/metrics/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/recall.py -------------------------------------------------------------------------------- /deepray/metrics/streaming_correlations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/streaming_correlations.py -------------------------------------------------------------------------------- /deepray/metrics/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/metrics/tests/cohens_kappa_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/cohens_kappa_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/f_scores_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/f_scores_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/geometric_mean_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/geometric_mean_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/hamming_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/hamming_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/harmonic_mean_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/harmonic_mean_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/metrics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/metrics_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/r_square_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/r_square_test.py -------------------------------------------------------------------------------- /deepray/metrics/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/metrics/utils.py -------------------------------------------------------------------------------- /deepray/models/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/BUILD -------------------------------------------------------------------------------- /deepray/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/README.md -------------------------------------------------------------------------------- /deepray/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/__init__.py -------------------------------------------------------------------------------- /deepray/models/albert_transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/albert_transformer_encoder.py -------------------------------------------------------------------------------- /deepray/models/bert_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/bert_classifier.py -------------------------------------------------------------------------------- /deepray/models/bert_pretrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/bert_pretrainer.py -------------------------------------------------------------------------------- /deepray/models/bert_span_labeler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/bert_span_labeler.py -------------------------------------------------------------------------------- /deepray/models/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/classification.py -------------------------------------------------------------------------------- /deepray/models/deep_cross_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/deep_cross_net.py -------------------------------------------------------------------------------- /deepray/models/encoder_scaffold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/encoder_scaffold.py -------------------------------------------------------------------------------- /deepray/models/framework.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/framework.py -------------------------------------------------------------------------------- /deepray/models/generative/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/models/generative/conditional_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/generative/conditional_gan.py -------------------------------------------------------------------------------- /deepray/models/masked_lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/masked_lm.py -------------------------------------------------------------------------------- /deepray/models/ncf_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/ncf_common.py -------------------------------------------------------------------------------- /deepray/models/ncf_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/ncf_model.py -------------------------------------------------------------------------------- /deepray/models/ncf_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/ncf_test.py -------------------------------------------------------------------------------- /deepray/models/neumf_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/neumf_model.py -------------------------------------------------------------------------------- /deepray/models/rec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/models/rec/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/deepfm.py -------------------------------------------------------------------------------- /deepray/models/rec/dien_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/dien_model.py -------------------------------------------------------------------------------- /deepray/models/rec/din_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/din_model.py -------------------------------------------------------------------------------- /deepray/models/rec/dual_channels_deep_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/dual_channels_deep_model.py -------------------------------------------------------------------------------- /deepray/models/rec/flen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/flen.py -------------------------------------------------------------------------------- /deepray/models/rec/flend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/flend.py -------------------------------------------------------------------------------- /deepray/models/rec/sim_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/rec/sim_model.py -------------------------------------------------------------------------------- /deepray/models/span_labeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/span_labeling.py -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/README.md -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/__init__.py -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/clip_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/clip_tokenizer.py -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/constants.py -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/decoder.py -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/image_encoder.py -------------------------------------------------------------------------------- /deepray/models/stable_diffusion/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/stable_diffusion/text_encoder.py -------------------------------------------------------------------------------- /deepray/models/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/models/tests/bert_classifier_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/bert_classifier_test.py -------------------------------------------------------------------------------- /deepray/models/tests/bert_pretrainer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/bert_pretrainer_test.py -------------------------------------------------------------------------------- /deepray/models/tests/bert_span_labeler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/bert_span_labeler_test.py -------------------------------------------------------------------------------- /deepray/models/tests/classification_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/classification_test.py -------------------------------------------------------------------------------- /deepray/models/tests/encoder_scaffold_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/encoder_scaffold_test.py -------------------------------------------------------------------------------- /deepray/models/tests/masked_lm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/masked_lm_test.py -------------------------------------------------------------------------------- /deepray/models/tests/span_labeling_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/span_labeling_test.py -------------------------------------------------------------------------------- /deepray/models/tests/transformer_encoder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/tests/transformer_encoder_test.py -------------------------------------------------------------------------------- /deepray/models/transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/transformer_encoder.py -------------------------------------------------------------------------------- /deepray/models/word2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/models/word2vec.py -------------------------------------------------------------------------------- /deepray/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/__init__.py -------------------------------------------------------------------------------- /deepray/ops/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/beam_search.py -------------------------------------------------------------------------------- /deepray/ops/beam_search_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/beam_search_test.py -------------------------------------------------------------------------------- /deepray/ops/decoding_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/decoding_module.py -------------------------------------------------------------------------------- /deepray/ops/decoding_module_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/decoding_module_test.py -------------------------------------------------------------------------------- /deepray/ops/sampling_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/sampling_module.py -------------------------------------------------------------------------------- /deepray/ops/segment_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/segment_extractor.py -------------------------------------------------------------------------------- /deepray/ops/segment_extractor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/ops/segment_extractor_test.py -------------------------------------------------------------------------------- /deepray/optimizers/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/BUILD -------------------------------------------------------------------------------- /deepray/optimizers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/README.md -------------------------------------------------------------------------------- /deepray/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/__init__.py -------------------------------------------------------------------------------- /deepray/optimizers/adabelief.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/adabelief.py -------------------------------------------------------------------------------- /deepray/optimizers/adagrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/adagrad.py -------------------------------------------------------------------------------- /deepray/optimizers/adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/adam.py -------------------------------------------------------------------------------- /deepray/optimizers/adam_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/adam_async.py -------------------------------------------------------------------------------- /deepray/optimizers/average_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/average_wrapper.py -------------------------------------------------------------------------------- /deepray/optimizers/cocob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/cocob.py -------------------------------------------------------------------------------- /deepray/optimizers/conditional_gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/conditional_gradient.py -------------------------------------------------------------------------------- /deepray/optimizers/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/constants.py -------------------------------------------------------------------------------- /deepray/optimizers/cyclical_learning_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/cyclical_learning_rate.py -------------------------------------------------------------------------------- /deepray/optimizers/ev_optimizer_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/ev_optimizer_patch.py -------------------------------------------------------------------------------- /deepray/optimizers/ftrl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/ftrl.py -------------------------------------------------------------------------------- /deepray/optimizers/gradient_descent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/gradient_descent.py -------------------------------------------------------------------------------- /deepray/optimizers/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/lamb.py -------------------------------------------------------------------------------- /deepray/optimizers/lazy_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/lazy_adam.py -------------------------------------------------------------------------------- /deepray/optimizers/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/lookahead.py -------------------------------------------------------------------------------- /deepray/optimizers/moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/moving_average.py -------------------------------------------------------------------------------- /deepray/optimizers/multi_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/multi_optimizer.py -------------------------------------------------------------------------------- /deepray/optimizers/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/novograd.py -------------------------------------------------------------------------------- /deepray/optimizers/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/optimization.py -------------------------------------------------------------------------------- /deepray/optimizers/proximal_adagrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/proximal_adagrad.py -------------------------------------------------------------------------------- /deepray/optimizers/rectified_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/rectified_adam.py -------------------------------------------------------------------------------- /deepray/optimizers/stochastic_weight_averaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/stochastic_weight_averaging.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/optimizers/tests/adabelief_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/adabelief_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/adam_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/adam_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/cocob_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/cocob_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/lamb_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/lamb_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/lazy_adam_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/lazy_adam_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/lookahead_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/lookahead_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/moving_average_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/moving_average_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/multi_optimizer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/multi_optimizer_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/novograd_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/novograd_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/proximal_adagrad_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/proximal_adagrad_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/rectified_adam_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/rectified_adam_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/standard_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/standard_test.py -------------------------------------------------------------------------------- /deepray/optimizers/tests/yogi_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/tests/yogi_test.py -------------------------------------------------------------------------------- /deepray/optimizers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/utils.py -------------------------------------------------------------------------------- /deepray/optimizers/warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/warmup.py -------------------------------------------------------------------------------- /deepray/optimizers/weight_decay_optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/weight_decay_optimizers.py -------------------------------------------------------------------------------- /deepray/optimizers/yogi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/optimizers/yogi.py -------------------------------------------------------------------------------- /deepray/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/options.py -------------------------------------------------------------------------------- /deepray/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/register.py -------------------------------------------------------------------------------- /deepray/testing/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/testing/BUILD -------------------------------------------------------------------------------- /deepray/testing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/testing/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/testing/serialization.py -------------------------------------------------------------------------------- /deepray/testing/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/testing/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/testing/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/testing/tests/serialization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/testing/tests/serialization_test.py -------------------------------------------------------------------------------- /deepray/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/tests/register_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/tests/register_test.py -------------------------------------------------------------------------------- /deepray/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/text/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/BUILD -------------------------------------------------------------------------------- /deepray/text/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/README.md -------------------------------------------------------------------------------- /deepray/text/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/__init__.py -------------------------------------------------------------------------------- /deepray/text/crf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/crf.py -------------------------------------------------------------------------------- /deepray/text/crf_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/crf_wrapper.py -------------------------------------------------------------------------------- /deepray/text/parse_time_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/parse_time_op.py -------------------------------------------------------------------------------- /deepray/text/skip_gram_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/skip_gram_ops.py -------------------------------------------------------------------------------- /deepray/text/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/text/tests/crf_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/tests/crf_test.py -------------------------------------------------------------------------------- /deepray/text/tests/crf_wrapper_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/tests/crf_wrapper_test.py -------------------------------------------------------------------------------- /deepray/text/tests/parse_time_op_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/tests/parse_time_op_test.py -------------------------------------------------------------------------------- /deepray/text/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/text/tests/skip_gram_ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/text/tests/skip_gram_ops_test.py -------------------------------------------------------------------------------- /deepray/utils/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/BUILD -------------------------------------------------------------------------------- /deepray/utils/README.md: -------------------------------------------------------------------------------- 1 | # Deepray Utils 2 | -------------------------------------------------------------------------------- /deepray/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/utils/accelerator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/utils/accelerator/tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/accelerator/tpu.py -------------------------------------------------------------------------------- /deepray/utils/accelerator/tpu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/accelerator/tpu_test.py -------------------------------------------------------------------------------- /deepray/utils/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/benchmark.py -------------------------------------------------------------------------------- /deepray/utils/ckpt_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/ckpt_util.py -------------------------------------------------------------------------------- /deepray/utils/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/utils/data/feature_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/data/feature_map.py -------------------------------------------------------------------------------- /deepray/utils/data/feature_map_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/data/feature_map_test.py -------------------------------------------------------------------------------- /deepray/utils/data/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/data/file_io.py -------------------------------------------------------------------------------- /deepray/utils/data/file_io_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/data/file_io_test.py -------------------------------------------------------------------------------- /deepray/utils/data/input_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/data/input_meta.py -------------------------------------------------------------------------------- /deepray/utils/ensure_tf_install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/ensure_tf_install.py -------------------------------------------------------------------------------- /deepray/utils/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/export/__init__.py -------------------------------------------------------------------------------- /deepray/utils/export/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/export/export.py -------------------------------------------------------------------------------- /deepray/utils/export/export_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/export/export_test.py -------------------------------------------------------------------------------- /deepray/utils/flags/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/README.md -------------------------------------------------------------------------------- /deepray/utils/flags/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/__init__.py -------------------------------------------------------------------------------- /deepray/utils/flags/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_base.py -------------------------------------------------------------------------------- /deepray/utils/flags/_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_benchmark.py -------------------------------------------------------------------------------- /deepray/utils/flags/_conventions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_conventions.py -------------------------------------------------------------------------------- /deepray/utils/flags/_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_data.py -------------------------------------------------------------------------------- /deepray/utils/flags/_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_device.py -------------------------------------------------------------------------------- /deepray/utils/flags/_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_distribution.py -------------------------------------------------------------------------------- /deepray/utils/flags/_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_misc.py -------------------------------------------------------------------------------- /deepray/utils/flags/_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/_performance.py -------------------------------------------------------------------------------- /deepray/utils/flags/common_flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/common_flags.py -------------------------------------------------------------------------------- /deepray/utils/flags/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/core.py -------------------------------------------------------------------------------- /deepray/utils/flags/flags_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/flags_test.py -------------------------------------------------------------------------------- /deepray/utils/flags/guidelines.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/flags/guidelines.md -------------------------------------------------------------------------------- /deepray/utils/gpu_affinity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/gpu_affinity.py -------------------------------------------------------------------------------- /deepray/utils/horovod_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/horovod_utils.py -------------------------------------------------------------------------------- /deepray/utils/keras_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/keras_utils.py -------------------------------------------------------------------------------- /deepray/utils/logging_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logging_util.py -------------------------------------------------------------------------------- /deepray/utils/logs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/utils/logs/cloud_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/cloud_lib.py -------------------------------------------------------------------------------- /deepray/utils/logs/cloud_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/cloud_lib_test.py -------------------------------------------------------------------------------- /deepray/utils/logs/guidelines.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/guidelines.md -------------------------------------------------------------------------------- /deepray/utils/logs/hooks_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/hooks_helper.py -------------------------------------------------------------------------------- /deepray/utils/logs/hooks_helper_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/hooks_helper_test.py -------------------------------------------------------------------------------- /deepray/utils/logs/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/logger.py -------------------------------------------------------------------------------- /deepray/utils/logs/logger_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/logger_test.py -------------------------------------------------------------------------------- /deepray/utils/logs/mlperf_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/mlperf_helper.py -------------------------------------------------------------------------------- /deepray/utils/logs/summary_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/logs/summary_manager.py -------------------------------------------------------------------------------- /deepray/utils/misc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/misc/__init__.py -------------------------------------------------------------------------------- /deepray/utils/misc/model_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/misc/model_helpers.py -------------------------------------------------------------------------------- /deepray/utils/misc/model_helpers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/misc/model_helpers_test.py -------------------------------------------------------------------------------- /deepray/utils/resource_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/resource_loader.py -------------------------------------------------------------------------------- /deepray/utils/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/test_utils.py -------------------------------------------------------------------------------- /deepray/utils/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepray/utils/tests/keras_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/tests/keras_utils_test.py -------------------------------------------------------------------------------- /deepray/utils/tests/run_all_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/tests/run_all_test.py -------------------------------------------------------------------------------- /deepray/utils/tests/test_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/tests/test_utils_test.py -------------------------------------------------------------------------------- /deepray/utils/tfrecord_inspecter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/tfrecord_inspecter.py -------------------------------------------------------------------------------- /deepray/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/timer.py -------------------------------------------------------------------------------- /deepray/utils/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/utils/types.py -------------------------------------------------------------------------------- /deepray/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/version.py -------------------------------------------------------------------------------- /deepray/workspace0.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/workspace0.bzl -------------------------------------------------------------------------------- /deepray/workspace2.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/workspace2.bzl -------------------------------------------------------------------------------- /deepray/workspace3.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/deepray/workspace3.bzl -------------------------------------------------------------------------------- /docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docker.sh -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/overview.md -------------------------------------------------------------------------------- /docs/tutorials/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/README.md -------------------------------------------------------------------------------- /docs/tutorials/_template.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/_template.ipynb -------------------------------------------------------------------------------- /docs/tutorials/_toc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/_toc.yaml -------------------------------------------------------------------------------- /docs/tutorials/average_optimizers_callback.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/average_optimizers_callback.ipynb -------------------------------------------------------------------------------- /docs/tutorials/image_ops.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/image_ops.ipynb -------------------------------------------------------------------------------- /docs/tutorials/layers_normalizations.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/layers_normalizations.ipynb -------------------------------------------------------------------------------- /docs/tutorials/layers_weightnormalization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/layers_weightnormalization.ipynb -------------------------------------------------------------------------------- /docs/tutorials/losses_triplet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/losses_triplet.ipynb -------------------------------------------------------------------------------- /docs/tutorials/networks_seq2seq_nmt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/networks_seq2seq_nmt.ipynb -------------------------------------------------------------------------------- /docs/tutorials/optimizers_lazyadam.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/optimizers_lazyadam.ipynb -------------------------------------------------------------------------------- /docs/tutorials/time_stopping.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/time_stopping.ipynb -------------------------------------------------------------------------------- /docs/tutorials/tqdm_progress_bar.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/docs/tutorials/tqdm_progress_bar.ipynb -------------------------------------------------------------------------------- /modelzoo/CV/Classify_images_of_clothing/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/Classify_images_of_clothing/train.py -------------------------------------------------------------------------------- /modelzoo/CV/GAN/conditional_gan_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/GAN/conditional_gan_mnist.py -------------------------------------------------------------------------------- /modelzoo/CV/GAN/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/GAN/train.py -------------------------------------------------------------------------------- /modelzoo/CV/SwinTransformers/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/SwinTransformers/model.py -------------------------------------------------------------------------------- /modelzoo/CV/SwinTransformers/run_horovod.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/SwinTransformers/run_horovod.sh -------------------------------------------------------------------------------- /modelzoo/CV/SwinTransformers/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/SwinTransformers/train.py -------------------------------------------------------------------------------- /modelzoo/CV/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/CV/mnist/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/CV/mnist/run_early.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/mnist/run_early.sh -------------------------------------------------------------------------------- /modelzoo/CV/mnist/run_horovod.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/mnist/run_horovod.sh -------------------------------------------------------------------------------- /modelzoo/CV/mnist/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/CV/mnist/train.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/.dockerignore -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/.gitignore -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/Dockerfile -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/README.md -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/bert_dllogger.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/bert_dllogger.json -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/common_flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/common_flags.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/data/Downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/data/Downloader.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/data/README.md -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/data/__init__.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/data/bertPrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/data/bertPrep.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/gpu_affinity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/gpu_affinity.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/input_pipeline.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/official/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/official/modeling/hyperparams/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/official/modeling/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/official/nlp/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/official/nlp/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/optimization.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/run.sub: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/run.sub -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/run_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/run_classifier.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/run_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/run_pretraining.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/run_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/run_squad.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/squad_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/squad_lib.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/squad_lib_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/squad_lib_sp.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/tf_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/tf_trt.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/BERT/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/BERT/tokenization.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/.gitignore -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/Dockerfile -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/LICENSE -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/NOTICE -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/README.md -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/file_utils.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/gpu_affinity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/gpu_affinity.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/modeling.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/optimization.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/run.sub: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/run.sub -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/run_tf_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/run_tf_squad.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/scripts/bind.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/scripts/bind.sh -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/squad_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/squad_utils.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/tokenization.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/utils.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/ELECTRA/vocab/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/ELECTRA/vocab/vocab.txt -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/Word2Vec/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/Word2Vec/dataset.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/Word2Vec/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/Word2Vec/model.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/Word2Vec/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/Word2Vec/run.sh -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/Word2Vec/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/Word2Vec/train.py -------------------------------------------------------------------------------- /modelzoo/LanguageModeling/Word2Vec/word2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/LanguageModeling/Word2Vec/word2vec.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/Criteo_DCN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/Criteo_DCN/README.md -------------------------------------------------------------------------------- /modelzoo/Recommendation/Criteo_DCN/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/Recommendation/Criteo_DCN/dcn_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/Criteo_DCN/dcn_v2.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/Criteo_DCN/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/Criteo_DCN/run.sh -------------------------------------------------------------------------------- /modelzoo/Recommendation/Criteo_DCN/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/Criteo_DCN/train.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/Criteo_DCN/train1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/Criteo_DCN/train1.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/MovieLens/mymodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/MovieLens/mymodel.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/MovieLens/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/MovieLens/run.sh -------------------------------------------------------------------------------- /modelzoo/Recommendation/MovieLens/run_ranking.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/MovieLens/run_ranking.sh -------------------------------------------------------------------------------- /modelzoo/Recommendation/MovieLens/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/MovieLens/train.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/NCF/run_ncf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/NCF/run_ncf.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/NCF/run_ncf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/NCF/run_ncf.sh -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/defaults.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/feature_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/feature_spec.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/main.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/run_dien.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/run_dien.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/run_din.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/run_din.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/run_horovod.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/run_horovod.sh -------------------------------------------------------------------------------- /modelzoo/Recommendation/SIM/run_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/SIM/run_sim.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/WideDeep/model.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/Recommendation/WideDeep/run_horovod.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/WideDeep/run_horovod.sh -------------------------------------------------------------------------------- /modelzoo/Recommendation/WideDeep/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/WideDeep/train.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/Recommendation/avazu-ctr-prediction/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/Recommendation/ev_test/ev_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/ev_test/ev_test.py -------------------------------------------------------------------------------- /modelzoo/Recommendation/ev_test/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/Recommendation/ev_test/run.sh -------------------------------------------------------------------------------- /modelzoo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modelzoo/horovod_test/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/modelzoo/horovod_test/run.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/setup.py -------------------------------------------------------------------------------- /third_party/BUILD: -------------------------------------------------------------------------------- 1 | licenses(["notice"]) # Apache 2.0 2 | -------------------------------------------------------------------------------- /third_party/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/arrow/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/arrow/arrow-20.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/arrow/arrow-20.patch -------------------------------------------------------------------------------- /third_party/arrow/arrow.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/arrow/arrow.BUILD -------------------------------------------------------------------------------- /third_party/aws-c-common.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/aws-c-common.BUILD -------------------------------------------------------------------------------- /third_party/aws-c-event-stream.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/aws-c-event-stream.BUILD -------------------------------------------------------------------------------- /third_party/aws-checksums.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/aws-checksums.BUILD -------------------------------------------------------------------------------- /third_party/aws-sdk-cpp.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/aws-sdk-cpp.BUILD -------------------------------------------------------------------------------- /third_party/brotli.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/brotli.BUILD -------------------------------------------------------------------------------- /third_party/clang_toolchain/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/clang_toolchain/download_clang.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/clang_toolchain/download_clang.bzl -------------------------------------------------------------------------------- /third_party/cuCollections/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/cuCollections/cuCollections.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/cuCollections/cuCollections.BUILD -------------------------------------------------------------------------------- /third_party/cuCollections/cuco.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/cuCollections/cuco.BUILD -------------------------------------------------------------------------------- /third_party/cuCollections/cucollection.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/cuCollections/cucollection.patch -------------------------------------------------------------------------------- /third_party/cutlass.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/cutlass.BUILD -------------------------------------------------------------------------------- /third_party/cxxtimer.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/cxxtimer.BUILD -------------------------------------------------------------------------------- /third_party/double_conversion.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/double_conversion.BUILD -------------------------------------------------------------------------------- /third_party/eigen3.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/eigen3.BUILD -------------------------------------------------------------------------------- /third_party/flash_attn/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/flash_attn/flash_attn.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/flash_attn/flash_attn.BUILD -------------------------------------------------------------------------------- /third_party/flash_attn/flash_attn.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/flash_attn/flash_attn.patch -------------------------------------------------------------------------------- /third_party/gpus/BUILD.bazel: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/gpus/find_cuda_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/gpus/find_cuda_config.py -------------------------------------------------------------------------------- /third_party/hadoop/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/hadoop/BUILD -------------------------------------------------------------------------------- /third_party/hadoop/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/hadoop/LICENSE.txt -------------------------------------------------------------------------------- /third_party/hadoop/hdfs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/hadoop/hdfs.h -------------------------------------------------------------------------------- /third_party/leveldb.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/leveldb.BUILD -------------------------------------------------------------------------------- /third_party/libcuckoo.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/libcuckoo.BUILD -------------------------------------------------------------------------------- /third_party/openblas.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/openblas.BUILD -------------------------------------------------------------------------------- /third_party/openssl.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/openssl.BUILD -------------------------------------------------------------------------------- /third_party/oss_c_sdk.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/oss_c_sdk.BUILD -------------------------------------------------------------------------------- /third_party/py/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/py/BUILD -------------------------------------------------------------------------------- /third_party/py/pypi.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/py/pypi.bzl -------------------------------------------------------------------------------- /third_party/py/python_init_pip.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/py/python_init_pip.bzl -------------------------------------------------------------------------------- /third_party/rapidjson.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/rapidjson.BUILD -------------------------------------------------------------------------------- /third_party/readerwriterqueue.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/readerwriterqueue.BUILD -------------------------------------------------------------------------------- /third_party/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/remote_config/common.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/remote_config/common.bzl -------------------------------------------------------------------------------- /third_party/sparsehash_c11/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/sparsehash_c11/sparsehash_c11.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/sparsehash_c11/sparsehash_c11.BUILD -------------------------------------------------------------------------------- /third_party/sparsehash_c11/sparsehash_c11.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/sparsehash_c11/sparsehash_c11.patch -------------------------------------------------------------------------------- /third_party/tf/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/tf/tf_215.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/tf/tf_215.patch -------------------------------------------------------------------------------- /third_party/thrift/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/thrift/build_defs.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/thrift/build_defs.bzl -------------------------------------------------------------------------------- /third_party/thrift/extra/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/thrift/extra/BUILD -------------------------------------------------------------------------------- /third_party/thrift/extra/config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/thrift/extra/config.h -------------------------------------------------------------------------------- /third_party/thrift/thrift.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/thrift/thrift.BUILD -------------------------------------------------------------------------------- /third_party/xla/BUILD.bazel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/xla/BUILD.bazel -------------------------------------------------------------------------------- /third_party/xla/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/xla/workspace.bzl -------------------------------------------------------------------------------- /third_party/xsimd.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/xsimd.BUILD -------------------------------------------------------------------------------- /third_party/xxHash.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/third_party/xxHash.BUILD -------------------------------------------------------------------------------- /tools/build_base_container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/build_base_container.sh -------------------------------------------------------------------------------- /tools/build_dev_container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/build_dev_container.sh -------------------------------------------------------------------------------- /tools/docker/base_container.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docker/base_container.Dockerfile -------------------------------------------------------------------------------- /tools/docker/bashrc.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docker/bashrc.bash -------------------------------------------------------------------------------- /tools/docker/build_wheel.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docker/build_wheel.Dockerfile -------------------------------------------------------------------------------- /tools/docker/dev_container.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docker/dev_container.Dockerfile -------------------------------------------------------------------------------- /tools/docker/entry.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docker/entry.sh -------------------------------------------------------------------------------- /tools/docker/pre-commit.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docker/pre-commit.Dockerfile -------------------------------------------------------------------------------- /tools/docs/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docs/BUILD -------------------------------------------------------------------------------- /tools/docs/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docs/Readme.md -------------------------------------------------------------------------------- /tools/docs/build_docs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/docs/build_docs.py -------------------------------------------------------------------------------- /tools/format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/format.py -------------------------------------------------------------------------------- /tools/install_deps/base.packages.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/base.packages.txt -------------------------------------------------------------------------------- /tools/install_deps/base.requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/base.requirements.txt -------------------------------------------------------------------------------- /tools/install_deps/buildifier.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/buildifier.sh -------------------------------------------------------------------------------- /tools/install_deps/clang-format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/clang-format.sh -------------------------------------------------------------------------------- /tools/install_deps/devel.packages.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/devel.packages.txt -------------------------------------------------------------------------------- /tools/install_deps/devel.requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/devel.requirements.txt -------------------------------------------------------------------------------- /tools/install_deps/doc_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/doc_requirements.txt -------------------------------------------------------------------------------- /tools/install_deps/install_bazelisk.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_bazelisk.sh -------------------------------------------------------------------------------- /tools/install_deps/install_clang.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_clang.sh -------------------------------------------------------------------------------- /tools/install_deps/install_cmake.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_cmake.sh -------------------------------------------------------------------------------- /tools/install_deps/install_miniforge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_miniforge.sh -------------------------------------------------------------------------------- /tools/install_deps/install_nsight-systems.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_nsight-systems.sh -------------------------------------------------------------------------------- /tools/install_deps/install_openmpi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_openmpi.sh -------------------------------------------------------------------------------- /tools/install_deps/install_openssh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_openssh.sh -------------------------------------------------------------------------------- /tools/install_deps/install_patchelf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_patchelf.sh -------------------------------------------------------------------------------- /tools/install_deps/install_python.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/install_python.sh -------------------------------------------------------------------------------- /tools/install_deps/setup.packages.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/install_deps/setup.packages.sh -------------------------------------------------------------------------------- /tools/pre-commit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/pre-commit.sh -------------------------------------------------------------------------------- /tools/releases/horovod_runner_patch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/releases/horovod_runner_patch.sh -------------------------------------------------------------------------------- /tools/releases/tf_auditwheel_patch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/releases/tf_auditwheel_patch.sh -------------------------------------------------------------------------------- /tools/run_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/run_build.sh -------------------------------------------------------------------------------- /tools/run_google_cloud_tests.sh: -------------------------------------------------------------------------------- 1 | set -x -e 2 | 3 | bash tools/run_gpu_tests.sh 4 | -------------------------------------------------------------------------------- /tools/run_gpu_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/run_gpu_tests.sh -------------------------------------------------------------------------------- /tools/testing/parallel_gpu_execute.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/testing/parallel_gpu_execute.sh -------------------------------------------------------------------------------- /tools/update_release_version.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepray-AI/deepray/HEAD/tools/update_release_version.sh --------------------------------------------------------------------------------