Hi @saykarthik, please feel free to take a look at Add AVX512 support in ATen & remove AVX support by imaginary-person · Pull Request #56992 · pytorch/pytorch · GitHub 49 which is in progress and which aims to add this support.  There are performance numbers planned to be added the PR before it’s ready to land (they might not be there yet).