vmap

PyTorch Developer Podcast - Een podcast door Edward Yang, Team PyTorch

Podcast artwork

Categorieën:

What is vmap? How is it implemented? How does our implementation compare to JAX's? What is a good way of understanding what vmap does? What's up with random numbers? Why are there some issues with the vmap that PyTorch currently ships?

Visit the podcast's native language site