LICENSE
MANIFEST.in
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/warnings.py
composer/algorithms/alibi/__init__.py
composer/algorithms/alibi/_gpt2_alibi.py
composer/algorithms/alibi/alibi.py
composer/algorithms/alibi/metadata.json
composer/algorithms/augmix/__init__.py
composer/algorithms/augmix/augmix.py
composer/algorithms/augmix/metadata.json
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/__init__.py
composer/algorithms/channels_last/channels_last.py
composer/algorithms/channels_last/metadata.json
composer/algorithms/colout/__init__.py
composer/algorithms/colout/colout.py
composer/algorithms/colout/metadata.json
composer/algorithms/cutmix/__init__.py
composer/algorithms/cutmix/cutmix.py
composer/algorithms/cutmix/metadata.json
composer/algorithms/cutout/__init__.py
composer/algorithms/cutout/cutout.py
composer/algorithms/cutout/metadata.json
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/__init__.py
composer/algorithms/ghost_batchnorm/ghost_batchnorm.py
composer/algorithms/ghost_batchnorm/metadata.json
composer/algorithms/label_smoothing/__init__.py
composer/algorithms/label_smoothing/label_smoothing.py
composer/algorithms/label_smoothing/metadata.json
composer/algorithms/layer_freezing/__init__.py
composer/algorithms/layer_freezing/layer_freezing.py
composer/algorithms/layer_freezing/metadata.json
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/__init__.py
composer/algorithms/progressive_resizing/metadata.json
composer/algorithms/progressive_resizing/progressive_resizing.py
composer/algorithms/randaugment/__init__.py
composer/algorithms/randaugment/metadata.json
composer/algorithms/randaugment/randaugment.py
composer/algorithms/sam/__init__.py
composer/algorithms/sam/metadata.json
composer/algorithms/sam/sam.py
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/checkpoint_saver.py
composer/callbacks/grad_monitor.py
composer/callbacks/lr_monitor.py
composer/callbacks/memory_monitor.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/datasets/__init__.py
composer/datasets/ade20k.py
composer/datasets/brats.py
composer/datasets/c4.py
composer/datasets/cifar.py
composer/datasets/coco.py
composer/datasets/dataloader.py
composer/datasets/dataset_registry.py
composer/datasets/evaluator.py
composer/datasets/ffcv_utils.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/synthetic_lm.py
composer/datasets/utils.py
composer/datasets/webdataset_utils.py
composer/functional/__init__.py
composer/loggers/__init__.py
composer/loggers/file_logger.py
composer/loggers/in_memory_logger.py
composer/loggers/logger.py
composer/loggers/logger_destination.py
composer/loggers/logger_hparams.py
composer/loggers/object_store_logger.py
composer/loggers/progress_bar_logger.py
composer/loggers/wandb_logger.py
composer/loss/__init__.py
composer/loss/loss.py
composer/loss/utils.py
composer/metrics/__init__.py
composer/metrics/metrics.py
composer/metrics/nlp.py
composer/models/__init__.py
composer/models/base.py
composer/models/initializers.py
composer/models/model_hparams.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/_layers.py
composer/models/efficientnetb0/efficientnetb0_hparams.py
composer/models/efficientnetb0/efficientnets.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/resnet_cifar/__init__.py
composer/models/resnet_cifar/model.py
composer/models/resnet_cifar/resnet_cifar_hparams.py
composer/models/resnet_cifar/resnets.py
composer/models/ssd/__init__.py
composer/models/ssd/base_model.py
composer/models/ssd/ssd.py
composer/models/ssd/ssd300.py
composer/models/ssd/ssd_hparams.py
composer/models/ssd/utils.py
composer/models/tasks/__init__.py
composer/models/tasks/classification.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/models/vit_small_patch16/__init__.py
composer/models/vit_small_patch16/hparams.py
composer/models/vit_small_patch16/model.py
composer/optim/__init__.py
composer/optim/decoupled_weight_decay.py
composer/optim/optimizer_hparams.py
composer/optim/scheduler.py
composer/optim/scheduler_hparams.py
composer/profiler/__init__.py
composer/profiler/dataloader_profiler.py
composer/profiler/json_trace_handler.py
composer/profiler/json_trace_merger.py
composer/profiler/marker.py
composer/profiler/profiler.py
composer/profiler/profiler_action.py
composer/profiler/profiler_hparams.py
composer/profiler/profiler_schedule.py
composer/profiler/system_profiler.py
composer/profiler/torch_profiler.py
composer/profiler/trace_handler.py
composer/trainer/__init__.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/checkpoint.py
composer/utils/dist.py
composer/utils/file_helpers.py
composer/utils/import_helpers.py
composer/utils/iter_helpers.py
composer/utils/module_surgery.py
composer/utils/object_store.py
composer/utils/reproducibility.py
composer/utils/string_enum.py
composer/yamls/algorithms/alibi.yaml
composer/yamls/algorithms/augmix.yaml
composer/yamls/algorithms/blurpool.yaml
composer/yamls/algorithms/channels_last.yaml
composer/yamls/algorithms/colout.yaml
composer/yamls/algorithms/curriculum_learning.yaml
composer/yamls/algorithms/cutmix.yaml
composer/yamls/algorithms/cutout.yaml
composer/yamls/algorithms/factorize.yaml
composer/yamls/algorithms/ghost_batchnorm.yaml
composer/yamls/algorithms/label_smoothing.yaml
composer/yamls/algorithms/layer_freezing.yaml
composer/yamls/algorithms/mixup.yaml
composer/yamls/algorithms/no_op_model.yaml
composer/yamls/algorithms/progressive_resizing.yaml
composer/yamls/algorithms/randaugment.yaml
composer/yamls/algorithms/sam.yaml
composer/yamls/algorithms/scale_schedule.yaml
composer/yamls/algorithms/selective_backprop.yaml
composer/yamls/algorithms/seq_length_warmup.yaml
composer/yamls/algorithms/squeeze_excite.yaml
composer/yamls/algorithms/stochastic_depth.yaml
composer/yamls/algorithms/swa.yaml
composer/yamls/algorithms/scale_schedule/0.25.yaml
composer/yamls/algorithms/scale_schedule/0.29.yaml
composer/yamls/algorithms/scale_schedule/0.33.yaml
composer/yamls/algorithms/scale_schedule/0.36.yaml
composer/yamls/algorithms/scale_schedule/0.43.yaml
composer/yamls/algorithms/scale_schedule/0.45.yaml
composer/yamls/algorithms/scale_schedule/0.50.yaml
composer/yamls/algorithms/scale_schedule/0.55.yaml
composer/yamls/algorithms/scale_schedule/0.56.yaml
composer/yamls/algorithms/scale_schedule/0.57.yaml
composer/yamls/algorithms/scale_schedule/0.64.yaml
composer/yamls/algorithms/scale_schedule/0.67.yaml
composer/yamls/algorithms/scale_schedule/0.71.yaml
composer/yamls/algorithms/scale_schedule/0.73.yaml
composer/yamls/algorithms/scale_schedule/0.75.yaml
composer/yamls/algorithms/scale_schedule/0.78.yaml
composer/yamls/algorithms/scale_schedule/0.79.yaml
composer/yamls/algorithms/scale_schedule/0.80.yaml
composer/yamls/algorithms/scale_schedule/0.82.yaml
composer/yamls/algorithms/scale_schedule/0.86.yaml
composer/yamls/algorithms/scale_schedule/0.89.yaml
composer/yamls/algorithms/scale_schedule/0.91.yaml
composer/yamls/algorithms/scale_schedule/0.93.yaml
composer/yamls/algorithms/scale_schedule/1.00.yaml
composer/yamls/algorithms/scale_schedule/1.25.yaml
composer/yamls/algorithms/scale_schedule/1.50.yaml
composer/yamls/algorithms/scale_schedule/2.00.yaml
composer/yamls/models/bert-base.yaml
composer/yamls/models/classify_mnist.yaml
composer/yamls/models/classify_mnist_cpu.yaml
composer/yamls/models/deeplabv3_ade20k_optimized.yaml
composer/yamls/models/deeplabv3_ade20k_unoptimized.yaml
composer/yamls/models/deeplabv3_wds_ade20k_optimized.yaml
composer/yamls/models/efficientnetb0.yaml
composer/yamls/models/gpt2_1,3b.yaml
composer/yamls/models/gpt2_125m.yaml
composer/yamls/models/gpt2_13b.yaml
composer/yamls/models/gpt2_2,7b.yaml
composer/yamls/models/gpt2_350m.yaml
composer/yamls/models/gpt2_52m.yaml
composer/yamls/models/gpt2_6,7b.yaml
composer/yamls/models/gpt2_760m.yaml
composer/yamls/models/gpt2_83m.yaml
composer/yamls/models/gpt3_125m.yaml
composer/yamls/models/gpt3_350m.yaml
composer/yamls/models/gpt3_760m.yaml
composer/yamls/models/resnet101.yaml
composer/yamls/models/resnet101_synthetic.yaml
composer/yamls/models/resnet152.yaml
composer/yamls/models/resnet152_synthetic.yaml
composer/yamls/models/resnet18.yaml
composer/yamls/models/resnet18_synthetic.yaml
composer/yamls/models/resnet20_cifar10.yaml
composer/yamls/models/resnet34.yaml
composer/yamls/models/resnet34_synthetic.yaml
composer/yamls/models/resnet50.yaml
composer/yamls/models/resnet50_synthetic.yaml
composer/yamls/models/resnet50_webdataset.yaml
composer/yamls/models/resnet56_cifar10.yaml
composer/yamls/models/resnet56_cifar10_synthetic.yaml
composer/yamls/models/resnet9_cifar10.yaml
composer/yamls/models/timm_resnet50_imagenet.yaml
composer/yamls/models/timm_vit_small_patch16.yaml
composer/yamls/models/unet.yaml
composer/yamls/models/vit_small_patch16.yaml
composer/yamls/models/glue/cola.yaml
composer/yamls/models/glue/mnli.yaml
composer/yamls/models/glue/mrpc.yaml
composer/yamls/models/glue/qnli.yaml
composer/yamls/models/glue/qqp.yaml
composer/yamls/models/glue/rte.yaml
composer/yamls/models/glue/sst-2.yaml
composer/yamls/models/glue/stsb.yaml
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