Gå frakoblet med Player FM -appen!
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
83 episoder
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
83 episoder
Alle episoder
×Velkommen til Player FM!
Player FM scanner netter for høykvalitets podcaster som du kan nyte nå. Det er den beste podcastappen og fungerer på Android, iPhone og internett. Registrer deg for å synkronisere abonnement på flere enheter.