LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
composer/__init__.py
composer/__main__.py
composer/_version.py
composer/py.typed
composer/algorithms/__init__.py
composer/algorithms/warnings.py
composer/algorithms/alibi/__init__.py
composer/algorithms/alibi/alibi.py
composer/algorithms/alibi/metadata.json
composer/algorithms/alibi/attention_surgery_functions/__init__.py
composer/algorithms/alibi/attention_surgery_functions/_bert.py
composer/algorithms/alibi/attention_surgery_functions/_gpt2.py
composer/algorithms/alibi/attention_surgery_functions/utils.py
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/ema/__init__.py
composer/algorithms/ema/ema.py
composer/algorithms/ema/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/fused_layernorm/__init__.py
composer/algorithms/fused_layernorm/fused_layernorm.py
composer/algorithms/fused_layernorm/metadata.json
composer/algorithms/gated_linear_units/__init__.py
composer/algorithms/gated_linear_units/gated_linear_unit_layers.py
composer/algorithms/gated_linear_units/gated_linear_units.py
composer/algorithms/gated_linear_units/metadata.json
composer/algorithms/ghost_batchnorm/__init__.py
composer/algorithms/ghost_batchnorm/ghost_batchnorm.py
composer/algorithms/ghost_batchnorm/metadata.json
composer/algorithms/gradient_clipping/__init__.py
composer/algorithms/gradient_clipping/gradient_clipping.py
composer/algorithms/gradient_clipping/metadata.json
composer/algorithms/gyro_dropout/__init__.py
composer/algorithms/gyro_dropout/gyro_dropout.py
composer/algorithms/gyro_dropout/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/low_precision_groupnorm/__init__.py
composer/algorithms/low_precision_groupnorm/low_precision_groupnorm.py
composer/algorithms/low_precision_groupnorm/metadata.json
composer/algorithms/low_precision_layernorm/__init__.py
composer/algorithms/low_precision_layernorm/low_precision_layernorm.py
composer/algorithms/low_precision_layernorm/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/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/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/algorithms/weight_standardization/__init__.py
composer/algorithms/weight_standardization/metadata.json
composer/algorithms/weight_standardization/weight_standardization.py
composer/callbacks/__init__.py
composer/callbacks/activation_monitor.py
composer/callbacks/checkpoint_saver.py
composer/callbacks/early_stopper.py
composer/callbacks/export_for_inference.py
composer/callbacks/generate.py
composer/callbacks/health_checker.py
composer/callbacks/image_visualizer.py
composer/callbacks/lr_monitor.py
composer/callbacks/memory_monitor.py
composer/callbacks/mlperf.py
composer/callbacks/optimizer_monitor.py
composer/callbacks/runtime_estimator.py
composer/callbacks/speed_monitor.py
composer/callbacks/system_metrics_monitor.py
composer/callbacks/threshold_stopper.py
composer/callbacks/utils.py
composer/cli/__init__.py
composer/cli/__main__.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/passes.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/ffcv_utils.py
composer/datasets/imagenet.py
composer/datasets/in_context_learning_evaluation.py
composer/datasets/lm_dataset.py
composer/datasets/mnist.py
composer/datasets/synthetic.py
composer/datasets/utils.py
composer/devices/__init__.py
composer/devices/device.py
composer/devices/device_cpu.py
composer/devices/device_gpu.py
composer/devices/device_hpu.py
composer/devices/device_mps.py
composer/devices/device_tpu.py
composer/functional/__init__.py
composer/loggers/__init__.py
composer/loggers/cometml_logger.py
composer/loggers/console_logger.py
composer/loggers/file_logger.py
composer/loggers/in_memory_logger.py
composer/loggers/logger.py
composer/loggers/logger_destination.py
composer/loggers/mlflow_logger.py
composer/loggers/mosaicml_logger.py
composer/loggers/progress_bar_logger.py
composer/loggers/remote_uploader_downloader.py
composer/loggers/slack_logger.py
composer/loggers/tensorboard_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/map.py
composer/metrics/metrics.py
composer/metrics/nlp.py
composer/models/__init__.py
composer/models/base.py
composer/models/huggingface.py
composer/models/initializers.py
composer/models/mmdetection.py
composer/models/bert/__init__.py
composer/models/bert/model.py
composer/models/classify_mnist/__init__.py
composer/models/classify_mnist/model.py
composer/models/deeplabv3/__init__.py
composer/models/deeplabv3/model.py
composer/models/efficientnetb0/__init__.py
composer/models/efficientnetb0/_layers.py
composer/models/efficientnetb0/efficientnets.py
composer/models/efficientnetb0/model.py
composer/models/gpt2/__init__.py
composer/models/gpt2/model.py
composer/models/resnet/__init__.py
composer/models/resnet/model.py
composer/models/resnet_cifar/__init__.py
composer/models/resnet_cifar/model.py
composer/models/resnet_cifar/resnets.py
composer/models/tasks/__init__.py
composer/models/tasks/classification.py
composer/models/timm/__init__.py
composer/models/timm/model.py
composer/models/unet/__init__.py
composer/models/unet/_layers.py
composer/models/unet/model.py
composer/models/unet/unet.py
composer/models/vit_small_patch16/__init__.py
composer/models/vit_small_patch16/model.py
composer/optim/__init__.py
composer/optim/decoupled_weight_decay.py
composer/optim/scheduler.py
composer/profiler/__init__.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_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/dist_strategy.py
composer/trainer/meta_safe_apply.py
composer/trainer/mosaic_fsdp.py
composer/trainer/mosaic_fsdp_utils.py
composer/trainer/trainer.py
composer/utils/__init__.py
composer/utils/auto_log_hparams.py
composer/utils/batch_helpers.py
composer/utils/checkpoint.py
composer/utils/collect_env.py
composer/utils/device.py
composer/utils/dist.py
composer/utils/file_helpers.py
composer/utils/fx_utils.py
composer/utils/import_helpers.py
composer/utils/inference.py
composer/utils/iter_helpers.py
composer/utils/misc.py
composer/utils/module_surgery.py
composer/utils/reproducibility.py
composer/utils/retrying.py
composer/utils/string_enum.py
composer/utils/eval_client/__init__.py
composer/utils/eval_client/eval_client.py
composer/utils/eval_client/lambda_eval_client.py
composer/utils/eval_client/local_eval_client.py
composer/utils/object_store/__init__.py
composer/utils/object_store/gcs_object_store.py
composer/utils/object_store/libcloud_object_store.py
composer/utils/object_store/object_store.py
composer/utils/object_store/oci_object_store.py
composer/utils/object_store/s3_object_store.py
composer/utils/object_store/sftp_object_store.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
tests/test_device.py
tests/test_docker.py
tests/test_docs.py
tests/test_engine.py
tests/test_events.py
tests/test_full_nlp.py
tests/test_loss.py
tests/test_notebooks.py
tests/test_passes.py
tests/test_precision.py
tests/test_simple_nlp.py
tests/test_smoketest.py
tests/test_split_batch.py
tests/test_state.py
tests/test_time.py