nntoolbox.callbacks.mixed_precision module

Implement mixed precision training as a callback Adapted from fastai’s course 2 v3 notebook and apex library https://arxiv.org/pdf/1710.03740.pdf