├── .gitignore ├── LICENSE ├── README.md ├── configs ├── ds_config_bf16_save_16bit_weights.json ├── ds_flan_t5_z3_config.json ├── ds_flan_t5_z3_config_bf16.json ├── ds_flan_t5_z3_offload.json └── ds_flan_t5_z3_offload_bf16.json ├── ds_launcher.py ├── requirements.txt ├── sagemaker-notebook.ipynb └── scripts └── run_seq2seq_deepspeed.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/README.md -------------------------------------------------------------------------------- /configs/ds_config_bf16_save_16bit_weights.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/configs/ds_config_bf16_save_16bit_weights.json -------------------------------------------------------------------------------- /configs/ds_flan_t5_z3_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/configs/ds_flan_t5_z3_config.json -------------------------------------------------------------------------------- /configs/ds_flan_t5_z3_config_bf16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/configs/ds_flan_t5_z3_config_bf16.json -------------------------------------------------------------------------------- /configs/ds_flan_t5_z3_offload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/configs/ds_flan_t5_z3_offload.json -------------------------------------------------------------------------------- /configs/ds_flan_t5_z3_offload_bf16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/configs/ds_flan_t5_z3_offload_bf16.json -------------------------------------------------------------------------------- /ds_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/ds_launcher.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/requirements.txt -------------------------------------------------------------------------------- /sagemaker-notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/sagemaker-notebook.ipynb -------------------------------------------------------------------------------- /scripts/run_seq2seq_deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philschmid/deepspeed-sagemaker-example/HEAD/scripts/run_seq2seq_deepspeed.py --------------------------------------------------------------------------------