rytse / torchemlp

Torch implementation of Marc Finzi's Equivariant MLP

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