/pytorch-fsdp-transformers

framework for plug and play of various transformers (vision and nlp) with FSDP

Primary LanguagePythonApache License 2.0Apache-2.0

transformer_framework

framework for plug and play of various transformers (vision and nlp) with FSDP