a6e7aa7796
* wip * setup * type * name * wip * docs * imports * fix if * fix if * use_amp * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * fix tests * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * fix tests * todos Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> |
||
---|---|---|
.. | ||
__init__.py | ||
cpu_backend.py | ||
ddp2_backend.py | ||
ddp_backend.py | ||
ddp_spawn_backend.py | ||
dp_backend.py | ||
gpu_backend.py | ||
tpu_backend.py |