Vmap PyTorch Developer podcast

Το περιεχόμενο παρέχεται από το PyTorch, Edward Yang, and Team PyTorch. Όλο το περιεχόμενο podcast, συμπεριλαμβανομένων των επεισοδίων, των γραφικών και των περιγραφών podcast, μεταφορτώνεται και παρέχεται απευθείας από τον PyTorch, Edward Yang, and Team PyTorch ή τον συνεργάτη της πλατφόρμας podcast. Εάν πιστεύετε ότι κάποιος χρησιμοποιεί το έργο σας που προστατεύεται από πνευματικά δικαιώματα χωρίς την άδειά σας, μπορείτε να ακολουθήσετε τη διαδικασία που περιγράφεται εδώ https://el.player.fm/legal.

PyTorch Developer Podcast « »
vmap

3+ y ago 17:47

Μοίρασέ το

MP3•Αρχική οθόνη επεισοδίου

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?

Further reading.

Tracking issue for vmap support https://github.com/pytorch/pytorch/issues/42368
BatchedTensor source code https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/BatchedTensorImpl.h , logical-physical transformation helper code https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/VmapTransforms.h (well documented, worth a read)
functorch, the better, more JAX-y implementation of vmap https://github.com/facebookresearch/functorch
Autodidax https://jax.readthedocs.io/en/latest/autodidax.html which contains a super simple vmap implementation that is a good model for the internal implementation that PyTorch has

83 επεισόδια

#Tech #PyTorch #Edward Yang #Team PyTorch #Deep Learning #Machine Learning

vmap

PyTorch Developer Podcast

33 subscribers

published 3+ y ago

Μοίρασέ το

MP3•Αρχική οθόνη επεισοδίου

Further reading.

Tracking issue for vmap support https://github.com/pytorch/pytorch/issues/42368
BatchedTensor source code https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/BatchedTensorImpl.h , logical-physical transformation helper code https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/VmapTransforms.h (well documented, worth a read)
functorch, the better, more JAX-y implementation of vmap https://github.com/facebookresearch/functorch
Autodidax https://jax.readthedocs.io/en/latest/autodidax.html which contains a super simple vmap implementation that is a good model for the internal implementation that PyTorch has

83 επεισόδια

#Tech #PyTorch #Edward Yang #Team PyTorch #Deep Learning #Machine Learning

Tous les épisodes

Καλώς ήλθατε στο Player FM!

Το FM Player σαρώνει τον ιστό για podcasts υψηλής ποιότητας για να απολαύσετε αυτή τη στιγμή. Είναι η καλύτερη εφαρμογή podcast και λειτουργεί σε Android, iPhone και στον ιστό. Εγγραφή για συγχρονισμό συνδρομών σε όλες τις συσκευές.

Ακούστε 500 + θέματα

Παρόμοιες με PyTorch Developer Podcast

Podcasts που αξίζει να ακούσετε

PyTorch Developer Podcast « » vmap

vmap

Podcasts που αξίζει να ακούσετε

Καλώς ήλθατε στο Player FM!

Παρόμοιες με PyTorch Developer Podcast

Οδηγός γρήγορης αναφοράς

PyTorch Developer Podcast « »
vmap