/torchemlp

Torch implementation of Marc Finzi's Equivariant MLP

Primary LanguagePythonMIT LicenseMIT

torchemlp

Torch implementation of Marc Finzi's Equivariant MLP.

Tools

Pyright

Before running pyright, generate the missing type stubs for libraries with

$ pyright --createstub matplotlib
$ pyright --createstub functorch

This will generate type stubs in the gitignored typings/ directory. After this step, you can run static analysis on the entire torchemlp library with

$ pyright