Gå offline med appen Player FM !
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 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
82 episoder
Alle episoder
×Välkommen till Player FM
Player FM scannar webben för högkvalitativa podcasts för dig att njuta av nu direkt. Den är den bästa podcast-appen och den fungerar med Android, Iphone och webben. Bli medlem för att synka prenumerationer mellan enheter.