![]() * add basic accelerator class. Co-Authored with @awaelchi * add basic trainign type plugin. Co-Authored with @awaelchi * pep8 Co-authored-by: @awaelchi * update copyright Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * add apex_amp Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * add mixed base class Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * add native amp Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * add native amp sharded Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * add tpu bfloat Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * add inits Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * Update precision_plugin.py Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> |
||
---|---|---|
.. | ||
legacy | ||
__init__.py | ||
accelerator.py |