PyTorch Developer Podcast

Vectorization

Listen Later

What is vectorization? How do you use it in PyTorch? What are some of the traps and pitfalls of writing vectorized code in PyTorch?

Further reading:

  • native/cpu README https://github.com/.../aten/src/ATen/native/cpu/README.md
  • Vec256 classes https://github.com/.../tree/master/aten/src/ATen/cpu/vec256
  • AVX512 support tracking issue https://github.com/pytorch/pytorch/issues/56187
...more
View all episodesView all episodes
Download on the App Store
PyTorch Developer PodcastBy Edward Yang, Team PyTorch
  • 4.8
  • 4.8
  • 4.8
  • 4.8
  • 4.8

4.8

49 ratings

More shows like PyTorch Developer Podcast

View all
Talk Python To Me by Michael Kennedy

Talk Python To Me

583 Listeners

Science Weekly by The Guardian

Science Weekly

429 Listeners

Cautionary Tales with Tim Harford by Pushkin Industries

Cautionary Tales with Tim Harford

5,109 Listeners

All-In with Chamath, Jason, Sacks & Friedberg by All-In Podcast, LLC

All-In with Chamath, Jason, Sacks & Friedberg

10,254 Listeners