An optimizer for Auto-wrapping FSDP
pip install -r requirements.txt
We actively welcome your pull requests.
- Fork the repo and create your branch from
main
. - If you've added code that should be tested, add tests.
- If you've changed APIs, update the documentation.
- Ensure the test suite passes.
- Make sure your code lints (
pre-commit run --all-files
).