A

accellerate

https://github.com/lucidrains/pytorch-custom-utils Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new AI research