Siirry offline-tilaan Player FM avulla!
Functional modules
Manage episode 296821911 series 2921809
Functional modules are a proposed mechanism to take PyTorch's existing NN module API and transform it into a functional form, where all the parameters are explicit argument. Why would you want to do this? What does functorch have to do with it? How come PyTorch's existing APIs don't seem to need this? What are the design problems?
Further reading.
- Proposal in GitHub issues https://github.com/pytorch/pytorch/issues/49171
- Linen design in flax https://flax.readthedocs.io/en/latest/design_notes/linen_design_principles.html
82 jaksoa
Manage episode 296821911 series 2921809
Functional modules are a proposed mechanism to take PyTorch's existing NN module API and transform it into a functional form, where all the parameters are explicit argument. Why would you want to do this? What does functorch have to do with it? How come PyTorch's existing APIs don't seem to need this? What are the design problems?
Further reading.
- Proposal in GitHub issues https://github.com/pytorch/pytorch/issues/49171
- Linen design in flax https://flax.readthedocs.io/en/latest/design_notes/linen_design_principles.html
82 jaksoa
Kaikki jaksot
×Tervetuloa Player FM:n!
Player FM skannaa verkkoa löytääkseen korkealaatuisia podcasteja, joista voit nauttia juuri nyt. Se on paras podcast-sovellus ja toimii Androidilla, iPhonela, ja verkossa. Rekisteröidy sykronoidaksesi tilaukset laitteiden välillä.