search

mfinzi / equivariant-MLP

A library for programmatically generating equivariant layers through constraint solving
MIT License
251 stars 21 forks source link

integrate Haiku and pytorch compatibility #3

Closed mfinzi closed 3 years ago

mfinzi commented 3 years ago