Rohan Varma 957ea485c4 [FSDP/AC] checkpoint_wrapper acccept auto_wrap_policy (#102672) ... Some feedback for this API is that folks would like to use auto_wrap_policy similar to FSDP instead of having to adapt to the signature of ``check_fn``. Differential Revision: [D46340320](https://our.internmc.facebook.com/intern/diff/D46340320/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/102672 Approved by: https://github.com/awgu		2023-06-04 18:31:19 +00:00
..
__init__.py
checkpoint_wrapper.py	[FSDP/AC] checkpoint_wrapper acccept auto_wrap_policy (#102672)	2023-06-04 18:31:19 +00:00