Hey 👋, I've just created a thread for this issue on PyTorch-Ignite Discord where you can quickly talk to the community on the topic.

^{🤖 This comment was automatically posted by Discuss on Discord}

Hi, @vfdev-5 , could you please assign me to this issue?

@Gulin7 you can start working on it

@vfdev-5 I searched for the use of torch.cuda.amp.autocast(... and found nothing, except a reference in some mock tests.

You should search for autocast, for example:
https://github.com/pytorch/ignite/blob/master/ignite/engine/__init__.py#L188