LICENSE.txt
README.md
pyproject.toml
cornstarch/__init__.py
cornstarch/pipeline_template.py
cornstarch.egg-info/PKG-INFO
cornstarch.egg-info/SOURCES.txt
cornstarch.egg-info/dependency_links.txt
cornstarch.egg-info/requires.txt
cornstarch.egg-info/top_level.txt
cornstarch/kernel/__init__.py
cornstarch/kernel/attention.py
cornstarch/kernel/bitfield_attention.py
cornstarch/kernel/interface.py
cornstarch/models/__init__.py
cornstarch/models/evaclip/__init__.py
cornstarch/models/evaclip/configuration_evaclip.py
cornstarch/models/evaclip/modeling_evaclip.py
cornstarch/models/intern_vit/__init__.py
cornstarch/models/intern_vit/configuration_intern_vit.py
cornstarch/models/intern_vit/flash_attention.py
cornstarch/models/intern_vit/modeling_intern_vit.py
cornstarch/models/multimodal_language_model/__init__.py
cornstarch/models/multimodal_language_model/configuration_multimodal_language_model.py
cornstarch/models/multimodal_language_model/modeling_multimodal_language_model.py
cornstarch/models/multimodal_language_model/processing_multimodal_language_model.py
cornstarch/plugin/__init__.py
cornstarch/plugin/encoders_colocated_plugin/__init__.py
cornstarch/plugin/encoders_colocated_plugin/encoders_colocated_plugin.py
cornstarch/plugin/encoders_colocated_plugin/encoders_colocated_stage_manager.py
cornstarch/plugin/encoders_colocated_plugin/one_f_one_b.py
cornstarch/plugin/encoders_colocated_plugin/process_group_mesh.py
cornstarch/plugin/encoders_replicated_plugin/__init__.py
cornstarch/plugin/encoders_replicated_plugin/encoders_replicated_plugin.py
cornstarch/plugin/encoders_replicated_plugin/encoders_replicated_stage_manager.py
cornstarch/plugin/encoders_replicated_plugin/process_group_mesh.py
cornstarch/plugin/heterogeneous_parallel_plugin/__init__.py
cornstarch/plugin/heterogeneous_parallel_plugin/heterogeneous_dataloader.py
cornstarch/plugin/heterogeneous_parallel_plugin/heterogeneous_parallel_module.py
cornstarch/plugin/heterogeneous_parallel_plugin/heterogeneous_parallel_plugin.py
cornstarch/plugin/heterogeneous_parallel_plugin/heterogeneous_process_group_mesh.py
cornstarch/plugin/heterogeneous_parallel_plugin/heterogeneous_stage_manager.py
cornstarch/plugin/multimodal_parallel_plugin/__init__.py
cornstarch/plugin/multimodal_parallel_plugin/modal_parallel_plugin.py
cornstarch/plugin/multimodal_parallel_plugin/modal_process_group_mesh.py
cornstarch/plugin/multimodal_parallel_plugin/multimodal_1f1b.py
cornstarch/plugin/multimodal_parallel_plugin/multimodal_parallel_checkpoint_io.py
cornstarch/plugin/multimodal_parallel_plugin/multimodal_parallel_plugin.py
cornstarch/plugin/multimodal_parallel_plugin/multimodal_stage_manager.py
cornstarch/shardformer/__init__.py
cornstarch/shardformer/layers/__init__.py
cornstarch/shardformer/layers/context_parallel_bitfield_attention.py
cornstarch/shardformer/layers/utils.py
cornstarch/shardformer/modeling/__init__.py
cornstarch/shardformer/modeling/clip.py
cornstarch/shardformer/modeling/dinov2.py
cornstarch/shardformer/modeling/evaclip.py
cornstarch/shardformer/modeling/gemma.py
cornstarch/shardformer/modeling/gemma2.py
cornstarch/shardformer/modeling/intern_vit.py
cornstarch/shardformer/modeling/llama.py
cornstarch/shardformer/modeling/mistral.py
cornstarch/shardformer/modeling/mixtral.py
cornstarch/shardformer/modeling/multimodal.py
cornstarch/shardformer/modeling/phi3.py
cornstarch/shardformer/modeling/pixtral.py
cornstarch/shardformer/modeling/qwen2.py
cornstarch/shardformer/modeling/qwen2_audio.py
cornstarch/shardformer/modeling/qwen2_vision.py
cornstarch/shardformer/modeling/siglip.py
cornstarch/shardformer/modeling/vit.py
cornstarch/shardformer/modeling/whisper.py
cornstarch/shardformer/policies/__init__.py
cornstarch/shardformer/policies/auto_policy.py
cornstarch/shardformer/policies/bert.py
cornstarch/shardformer/policies/clip.py
cornstarch/shardformer/policies/dinov2.py
cornstarch/shardformer/policies/evaclip.py
cornstarch/shardformer/policies/gemma.py
cornstarch/shardformer/policies/gemma2.py
cornstarch/shardformer/policies/gpt2.py
cornstarch/shardformer/policies/intern_vit.py
cornstarch/shardformer/policies/llama.py
cornstarch/shardformer/policies/mistral.py
cornstarch/shardformer/policies/mixtral.py
cornstarch/shardformer/policies/multimodal.py
cornstarch/shardformer/policies/opt.py
cornstarch/shardformer/policies/phi3.py
cornstarch/shardformer/policies/pipeline_template_policy.py
cornstarch/shardformer/policies/pixtral.py
cornstarch/shardformer/policies/qwen2.py
cornstarch/shardformer/policies/qwen2_audio.py
cornstarch/shardformer/policies/qwen2_vision.py
cornstarch/shardformer/policies/siglip.py
cornstarch/shardformer/policies/vit.py
cornstarch/shardformer/policies/whisper.py
cornstarch/shardformer/shard/__init__.py
cornstarch/shardformer/shard/placeholder.py
cornstarch/shardformer/shard/shard_config.py
cornstarch/shardformer/shard/shardformer.py
cornstarch/testing/__init__.py
cornstarch/testing/dcgm.py
cornstarch/testing/model_zoo.py
cornstarch/testing/profile_4d_parallel.py
cornstarch/testing/profile_context_parallel.py
cornstarch/testing/profile_fsdp.py
cornstarch/testing/profile_layer.py
cornstarch/testing/profile_pipeline_parallel.py
cornstarch/testing/timer.py
cornstarch/testing/utils.py
tests/test_pipeline_template.py