search

e3nn / e3nn-jax

jax library for E3 Equivariant Neural Networks
Apache License 2.0
182 stars 18 forks source link

Add custom initializers and instructions in Linear. #78

Closed ameya98 closed 2 months ago

ameya98 commented 2 months ago

Should resolve both requests in #77!