LICENSE
README.md
setup.py
examples/__init__.py
examples/shared_args.py
examples/fashion_mnist/__init__.py
examples/fashion_mnist/shared_args.py
examples/fashion_mnist/pytorch/__init__.py
examples/fashion_mnist/pytorch/train.py
examples/fashion_mnist/pytorch/xla/__init__.py
examples/fashion_mnist/pytorch/xla/train.py
examples/fashion_mnist/tensorflow/__init__.py
examples/fashion_mnist/tensorflow/train.py
examples/hello_world/__init__.py
examples/hello_world/pytorch/__init__.py
examples/hello_world/pytorch/train.py
examples/hello_world/pytorch/train_eager.py
examples/hello_world/pytorch/xla/__init__.py
examples/hello_world/pytorch/xla/train.py
examples/hello_world/tensorflow/__init__.py
examples/hello_world/tensorflow/train.py
examples/hello_world/tensorflow/train_eager.py
examples/legacy/__init__.py
examples/legacy/chatbot/__init__.py
examples/legacy/chatbot/chatbot_data_reference.py
examples/legacy/chatbot/conversation_dataset.py
examples/legacy/chatbot/conversation_parsing.py
examples/legacy/chatbot/shared.py
examples/legacy/chatbot/split_conversations_dataset.py
examples/legacy/chatbot/voc.py
examples/legacy/chatbot/pytorch/__init__.py
examples/legacy/chatbot/pytorch/train_chatbot_reference.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/__init__.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/chatbot_reference.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/chatbot_reference_distributed.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/evaluate_reference_chatbot.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/evaluation.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/loss.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/model_data_generation.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/seq2seq.py
examples/legacy/chatbot/pytorch/original_chatbot_tutorial/training.py
examples/legacy/chatbot/tensorflow/__init__.py
examples/legacy/chatbot/tensorflow/train_chatbot_reference.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/__init__.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/chatbot_reference.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/evaluation.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/model_data_generation.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/training.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/transformer.py
examples/legacy/chatbot/tensorflow/original_transformer_tutorial/transformer_tutorial.py
examples/persona_chatbot/__init__.py
examples/persona_chatbot/shared_args.py
examples/persona_chatbot/special_tokens.py
examples/persona_chatbot/training_process.py
examples/persona_chatbot/datasets/__init__.py
examples/persona_chatbot/datasets/conversations.py
examples/persona_chatbot/datasets/distributed.py
examples/persona_chatbot/datasets/manager.py
examples/persona_chatbot/datasets/multiple_choice.py
examples/persona_chatbot/datasets/tokenizers.py
examples/persona_chatbot/pytorch/__init__.py
examples/persona_chatbot/pytorch/chat.py
examples/persona_chatbot/pytorch/collation.py
examples/persona_chatbot/pytorch/shared_args.py
examples/persona_chatbot/pytorch/train.py
examples/persona_chatbot/tensorflow/__init__.py
examples/persona_chatbot/tensorflow/collation.py
examples/persona_chatbot/tensorflow/train.py
experiments/__init__.py
experiments/tensorflow/__init__.py
experiments/tensorflow/accumulate_nested.py
experiments/tensorflow/tensor_slicing_on_device.py
experiments/tensorflow/test_accumulate_nested.py
experiments/tensorflow/test_distributed_chunked_processing.py
experiments/tensorflow/test_gather_deadlock.py
experiments/tensorflow/test_tf_batch_chunking.py
experiments/tensorflow/test_tf_batch_chunking_2.py
experiments/tensorflow/test_tf_dataset.py
experiments/tensorflow/test_tf_resize_token_size.py
issues/__init__.py
issues/compiled_train_step_with_lr_scheduler_feb_2024.py
mlpug/__init__.py
mlpug/abstract_interface.py
mlpug/base.py
mlpug/batch_chunking.py
mlpug/debugging.py
mlpug/evaluation.py
mlpug/mlpug_exceptions.py
mlpug/multi_processing.py
mlpug/reserved.py
mlpug/scheduler_funcs.py
mlpug.egg-info/PKG-INFO
mlpug.egg-info/SOURCES.txt
mlpug.egg-info/dependency_links.txt
mlpug.egg-info/requires.txt
mlpug.egg-info/top_level.txt
mlpug/mlpug_logging/__init__.py
mlpug/mlpug_logging/base.py
mlpug/mlpug_logging/colored_logging.py
mlpug/pytorch/__init__.py
mlpug/pytorch/batch_chunking.py
mlpug/pytorch/evaluation.py
mlpug/pytorch/multi_processing.py
mlpug/pytorch/scheduler_funcs.py
mlpug/pytorch/trainers/__init__.py
mlpug/pytorch/trainers/training.py
mlpug/pytorch/trainers/callbacks/__init__.py
mlpug/pytorch/trainers/callbacks/basic.py
mlpug/pytorch/trainers/callbacks/callback.py
mlpug/pytorch/trainers/callbacks/checkpoint_manager.py
mlpug/pytorch/trainers/callbacks/cuda_memory.py
mlpug/pytorch/trainers/callbacks/distributed.py
mlpug/pytorch/trainers/callbacks/lr_scheduler_wrapper.py
mlpug/pytorch/trainers/callbacks/metrics_logger.py
mlpug/pytorch/trainers/callbacks/tensorboard.py
mlpug/pytorch/utils/__init__.py
mlpug/pytorch/utils/mlpug_data.py
mlpug/pytorch/utils/sliding_window.py
mlpug/pytorch/xla/__init__.py
mlpug/pytorch/xla/batch_chunking.py
mlpug/pytorch/xla/evaluation.py
mlpug/pytorch/xla/multi_processing.py
mlpug/pytorch/xla/scheduler_funcs.py
mlpug/pytorch/xla/trainers/__init__.py
mlpug/pytorch/xla/trainers/training.py
mlpug/pytorch/xla/trainers/callbacks/__init__.py
mlpug/pytorch/xla/trainers/callbacks/basic.py
mlpug/pytorch/xla/trainers/callbacks/callback.py
mlpug/pytorch/xla/trainers/callbacks/checkpoint_manager.py
mlpug/pytorch/xla/trainers/callbacks/distributed.py
mlpug/pytorch/xla/trainers/callbacks/lr_scheduler_wrapper.py
mlpug/pytorch/xla/trainers/callbacks/metrics_logger.py
mlpug/pytorch/xla/trainers/callbacks/tensorboard.py
mlpug/tensorflow/__init__.py
mlpug/tensorflow/batch_chunking.py
mlpug/tensorflow/distributed_utils.py
mlpug/tensorflow/evaluation.py
mlpug/tensorflow/func_wrapping.py
mlpug/tensorflow/scheduler_funcs.py
mlpug/tensorflow/trainers/__init__.py
mlpug/tensorflow/trainers/training.py
mlpug/tensorflow/trainers/callbacks/__init__.py
mlpug/tensorflow/trainers/callbacks/basic.py
mlpug/tensorflow/trainers/callbacks/callback.py
mlpug/tensorflow/trainers/callbacks/checkpoint_manager.py
mlpug/tensorflow/trainers/callbacks/lr_scheduler_wrapper.py
mlpug/tensorflow/trainers/callbacks/metrics_logger.py
mlpug/tensorflow/trainers/callbacks/tensorboard.py
mlpug/trainers/__init__.py
mlpug/trainers/training.py
mlpug/trainers/callbacks/__init__.py
mlpug/trainers/callbacks/basic.py
mlpug/trainers/callbacks/callback.py
mlpug/trainers/callbacks/checkpoint_manager.py
mlpug/trainers/callbacks/lr_scheduler_wrapper.py
mlpug/trainers/callbacks/metrics_logger.py
mlpug/trainers/callbacks/tensorboard.py
mlpug/utils/__init__.py
mlpug/utils/mlpug_data.py
mlpug/utils/sliding_window.py
mlpug/utils/utils.py