CHANGELOG.md
CITATION.cff
LICENSE
MANIFEST.in
README.md
SECURITY.md
pyproject.toml
requirements.txt
setup.py
.actions/assistant.py
requirements/base.txt
requirements/cli.txt
requirements/devel.txt
requirements/docs.txt
requirements/examples.txt
requirements/extra.txt
requirements/ipynb.txt
requirements/lightning_pin.txt
requirements/standalone_base.txt
requirements/test.txt
src/finetuning_scheduler/__about__.py
src/finetuning_scheduler/__init__.py
src/finetuning_scheduler/fts.py
src/finetuning_scheduler/fts_supporters.py
src/finetuning_scheduler/py.typed
src/finetuning_scheduler/setup_tools.py
src/finetuning_scheduler/types.py
src/finetuning_scheduler.egg-info/PKG-INFO
src/finetuning_scheduler.egg-info/SOURCES.txt
src/finetuning_scheduler.egg-info/dependency_links.txt
src/finetuning_scheduler.egg-info/not-zip-safe
src/finetuning_scheduler.egg-info/requires.txt
src/finetuning_scheduler.egg-info/top_level.txt
src/finetuning_scheduler/strategy_adapters/__init__.py
src/finetuning_scheduler/strategy_adapters/_mp_imports.py
src/finetuning_scheduler/strategy_adapters/_wrap_utils.py
src/finetuning_scheduler/strategy_adapters/base.py
src/finetuning_scheduler/strategy_adapters/fsdp.py
src/finetuning_scheduler/strategy_adapters/model_parallel.py
src/fts_examples/__init__.py
src/fts_examples/cfg_utils.py
src/fts_examples/cli_experiment_utils.py
src/fts_examples/fts_superglue.py
src/fts_examples/config/RteBoolqModule_ft_schedule_deberta_base.yaml
src/fts_examples/config/RteBoolqModule_ft_schedule_deberta_base_fsdp.yaml
src/fts_examples/config/fts_defaults.yaml
src/fts_examples/config/fts_explicit.yaml
src/fts_examples/config/fts_implicit.yaml
src/fts_examples/config/nofts_baseline.yaml
src/fts_examples/config/advanced/fsdp/fts_ddp_fsdp_baseline_profile.yaml
src/fts_examples/config/advanced/fsdp/fts_fsdp_awp_overrides_offload_profile.yaml
src/fts_examples/config/advanced/fsdp/fts_fsdp_awp_overrides_profile.yaml
src/fts_examples/config/advanced/fsdp/fts_fsdp_basic_profile.yaml
src/fts_examples/config/advanced/reinit_lr/explicit_reinit_lr.yaml
src/fts_examples/config/advanced/reinit_lr/fts_explicit_reinit_lr.yaml
src/fts_examples/config/advanced/reinit_lr/fts_implicit_reinit_lr.yaml
src/fts_examples/config/advanced/reinit_optim_lr/explicit_reinit_optim_lr.yaml
src/fts_examples/config/advanced/reinit_optim_lr/fts_explicit_reinit_optim_lr.yaml
src/fts_examples/config/advanced/reinit_optim_lr/fts_implicit_reinit_optim_lr.yaml
src/fts_examples/config/advanced/reinit_optim_lr/fts_implicit_reinit_optim_lr_use_curr.yaml
src/fts_examples/model_parallel/mp_examples.py
src/fts_examples/model_parallel/torchtitan_llama.py
src/fts_examples/model_parallel/config/fts_fsdp_auto_plan.yaml
src/fts_examples/model_parallel/config/fts_fsdp_profiling.yaml
src/fts_examples/model_parallel/config/fts_tp_plan.yaml
src/fts_examples/model_parallel/config/defaults/fts_mp_example_defaults.yaml
src/fts_examples/model_parallel/config/defaults/llama_ft_schedule.yaml
src/fts_examples/model_parallel/config/defaults/torchtitan_llama.yaml
src/fts_examples/model_parallel/config/profiling/memprofiler_demo.yaml
src/fts_examples/model_parallel/config/profiling/pytorch_profiler.yaml
src/fts_examples/patching/__init__.py
src/fts_examples/patching/_patch_utils.py
src/fts_examples/patching/dep_patch_shim.py
src/fts_examples/patching/patched_einsum_strategies.py
src/fts_examples/patching/patched_lightning_jsonargparse.py
src/fts_examples/patching/patched_numpyarrowextractor.py
src/fts_examples/patching/patched_triton_jit_fn_init.py
src/fts_examples/profiling/__init__.py
src/fts_examples/profiling/config.py
src/fts_examples/profiling/extended_profiler.py
src/fts_examples/profiling/memprofiler.py
src/fts_examples/profiling/npp_hooks.py
src/fts_examples/profiling/profiler_hooks_mixin.py
tests/test_finetuning_scheduler_callback.py
tests/test_fsdp.py
tests/test_model_parallel.py