LICENSE
LICENSE_HEADER
README.md
pyproject.toml
setup.py
composer/__init__.py
composer/py.typed
composer/algorithms/__init__.py
composer/algorithms/algorithm_hparams.py
composer/algorithms/algorithm_registry.py
composer/algorithms/hparams.py
composer/algorithms/alibi/README.md
composer/algorithms/alibi/__init__.py
composer/algorithms/alibi/_gpt2_alibi.py
composer/algorithms/alibi/alibi.py
composer/algorithms/alibi/metadata.json
composer/algorithms/augmix/README.md
composer/algorithms/augmix/__init__.py
composer/algorithms/augmix/augmix.py
composer/algorithms/augmix/metadata.json
composer/algorithms/blurpool/README.md
composer/algorithms/blurpool/__init__.py
composer/algorithms/blurpool/blurpool.py
composer/algorithms/blurpool/blurpool_layers.py
composer/algorithms/blurpool/metadata.json
composer/algorithms/channels_last/README.md
composer/algorithms/channels_last/__init__.py
composer/algorithms/channels_last/channels_last.py
composer/algorithms/channels_last/metadata.json
composer/algorithms/colout/README.md
composer/algorithms/colout/__init__.py
composer/algorithms/colout/colout.py
composer/algorithms/colout/metadata.json
composer/algorithms/cutmix/README.md
composer/algorithms/cutmix/__init__.py
composer/algorithms/cutmix/cutmix.py
composer/algorithms/cutmix/metadata.json
composer/algorithms/cutout/README.md
composer/algorithms/cutout/__init__.py
composer/algorithms/cutout/cutout.py
composer/algorithms/cutout/metadata.json
composer/algorithms/factorize/README.md
composer/algorithms/factorize/__init__.py
composer/algorithms/factorize/factorize.py
composer/algorithms/factorize/factorize_core.py
composer/algorithms/factorize/factorize_modules.py
composer/algorithms/factorize/metadata.json
composer/algorithms/ghost_batchnorm/README.md
composer/algorithms/ghost_batchnorm/__init__.py
composer/algorithms/ghost_batchnorm/ghost_batchnorm.py
composer/algorithms/ghost_batchnorm/metadata.json
composer/algorithms/label_smoothing/README.md
composer/algorithms/label_smoothing/__init__.py
composer/algorithms/label_smoothing/label_smoothing.py
composer/algorithms/label_smoothing/metadata.json
composer/algorithms/layer_freezing/README.md
composer/algorithms/layer_freezing/__init__.py
composer/algorithms/layer_freezing/layer_freezing.py
composer/algorithms/layer_freezing/metadata.json
composer/algorithms/mixup/README.md
composer/algorithms/mixup/__init__.py
composer/algorithms/mixup/metadata.json
composer/algorithms/mixup/mixup.py
composer/algorithms/no_op_model/__init__.py
composer/algorithms/no_op_model/metadata.json
composer/algorithms/no_op_model/no_op_model.py
composer/algorithms/progressive_resizing/README.md
composer/algorithms/progressive_resizing/__init__.py
composer/algorithms/progressive_resizing/metadata.json
composer/algorithms/progressive_resizing/progressive_resizing.py
composer/algorithms/randaugment/README.md
composer/algorithms/randaugment/__init__.py
composer/algorithms/randaugment/metadata.json
composer/algorithms/randaugment/randaugment.py
composer/algorithms/sam/README.md
composer/algorithms/sam/__init__.py
composer/algorithms/sam/metadata.json
composer/algorithms/sam/sam.py
composer/algorithms/scale_schedule/README.md
composer/algorithms/scale_schedule/__init__.py
composer/algorithms/scale_schedule/metadata.json
composer/algorithms/scale_schedule/scale_schedule.py
composer/algorithms/selective_backprop/__init__.py
composer/algorithms/selective_backprop/metadata.json
composer/algorithms/selective_backprop/selective_backprop.py
composer/algorithms/seq_length_warmup/__init__.py
composer/algorithms/seq_length_warmup/metadata.json
composer/algorithms/seq_length_warmup/seq_length_warmup.py
composer/algorithms/squeeze_excite/__init__.py
composer/algorithms/squeeze_excite/metadata.json
composer/algorithms/squeeze_excite/squeeze_excite.py
composer/algorithms/stochastic_depth/__init__.py
composer/algorithms/stochastic_depth/metadata.json
composer/algorithms/stochastic_depth/sample_stochastic_layers.py
composer/algorithms/stochastic_depth/stochastic_depth.py
composer/algorithms/stochastic_depth/stochastic_layers.py
composer/algorithms/swa/__init__.py
composer/algorithms/swa/metadata.json
composer/algorithms/swa/swa.py
composer/algorithms/utils/__init__.py
composer/algorithms/utils/augmentation_common.py
composer/algorithms/utils/augmentation_primitives.py
composer/callbacks/__init__.py
composer/callbacks/callback_hparams.py
composer/callbacks/grad_monitor.py
composer/callbacks/lr_monitor.py
composer/callbacks/memory_monitor.py
composer/callbacks/run_directory_uploader.py
composer/callbacks/speed_monitor.py
composer/cli/__init__.py
composer/cli/launcher.py
composer/core/__init__.py
composer/core/algorithm.py
composer/core/callback.py
composer/core/data_spec.py
composer/core/engine.py
composer/core/evaluator.py
composer/core/event.py
composer/core/precision.py
composer/core/serializable.py
composer/core/state.py
composer/core/time.py
composer/core/types.py
composer/core/logging/__init__.py
composer/core/logging/base_backend.py
composer/core/logging/logger.py
composer/datasets/__init__.py
composer/datasets/ade20k.py
composer/datasets/brats.py
composer/datasets/cifar10.py
composer/datasets/dataloader.py
composer/datasets/dataset_registry.py
composer/datasets/evaluator.py
composer/datasets/glue.py
composer/datasets/hparams.py
composer/datasets/imagenet.py
composer/datasets/lm_datasets.py
composer/datasets/mnist.py
composer/datasets/synthetic.py
composer/datasets/utils.py
composer/functional/__init__.py
composer/loggers/__init__.py
composer/loggers/file_logger.py
composer/loggers/in_memory_logger.py
composer/loggers/logger_hparams.py
composer/loggers/registry.py
composer/loggers/tqdm_logger.py
composer/loggers/wandb_logger.py
composer/models/__init__.py
composer/models/base.py
composer/models/efficientnets.py
composer/models/initializers.py
composer/models/loss.py
composer/models/model_hparams.py
composer/models/nlp_metrics.py
composer/models/resnets.py
composer/models/transformer_hparams.py
composer/models/transformer_shared.py
composer/models/bert/__init__.py
composer/models/bert/bert_hparams.py
composer/models/bert/model.py
composer/models/classify_mnist/__init__.py
composer/models/classify_mnist/mnist_hparams.py
composer/models/classify_mnist/model.py
composer/models/deeplabv3/__init__.py
composer/models/deeplabv3/deeplabv3.py
composer/models/deeplabv3/deeplabv3_hparams.py
composer/models/efficientnetb0/__init__.py
composer/models/efficientnetb0/efficientnetb0_hparams.py
composer/models/efficientnetb0/model.py
composer/models/gpt2/__init__.py
composer/models/gpt2/gpt2_hparams.py
composer/models/gpt2/model.py
composer/models/resnet/__init__.py
composer/models/resnet/model.py
composer/models/resnet/resnet_hparams.py
composer/models/resnet20_cifar10/__init__.py
composer/models/resnet20_cifar10/model.py
composer/models/resnet20_cifar10/resnet20_cifar10_hparams.py
composer/models/resnet56_cifar10/__init__.py
composer/models/resnet56_cifar10/model.py
composer/models/resnet56_cifar10/resnet56_cifar10_hparams.py
composer/models/resnet9_cifar10/__init__.py
composer/models/resnet9_cifar10/model.py
composer/models/resnet9_cifar10/resnet9_cifar10_hparams.py
composer/models/timm/__init__.py
composer/models/timm/model.py
composer/models/timm/timm_hparams.py
composer/models/unet/__init__.py
composer/models/unet/layers.py
composer/models/unet/model.py
composer/models/unet/unet.py
composer/models/unet/unet_hparams.py
composer/optim/__init__.py
composer/optim/_time_conversion.py
composer/optim/decoupled_weight_decay.py
composer/optim/optimizer_hparams.py
composer/optim/scheduler.py
composer/profiler/__init__.py
composer/profiler/_event_handler.py
composer/profiler/_profiler.py
composer/profiler/_profiler_action.py
composer/profiler/dataloader_profiler.py
composer/profiler/json_trace.py
composer/profiler/json_trace_merger.py
composer/profiler/profiler_hparams.py
composer/profiler/system_profiler.py
composer/profiler/torch_profiler.py
composer/trainer/__init__.py
composer/trainer/_checkpoint.py
composer/trainer/_deepspeed.py
composer/trainer/_scale_schedule.py
composer/trainer/_scaler.py
composer/trainer/ddp.py
composer/trainer/trainer.py
composer/trainer/trainer_hparams.py
composer/trainer/devices/__init__.py
composer/trainer/devices/device.py
composer/trainer/devices/device_cpu.py
composer/trainer/devices/device_gpu.py
composer/trainer/devices/device_hparams.py
composer/utils/__init__.py
composer/utils/dist.py
composer/utils/iter_helpers.py
composer/utils/module_surgery.py
composer/utils/object_store.py
composer/utils/reproducibility.py
composer/utils/run_directory.py
composer/utils/string_enum.py
mosaicml.egg-info/PKG-INFO
mosaicml.egg-info/SOURCES.txt
mosaicml.egg-info/dependency_links.txt
mosaicml.egg-info/entry_points.txt
mosaicml.egg-info/requires.txt
mosaicml.egg-info/top_level.txt