Commit 89c18d9

committed

feat: add jit for so2 tensor product and E3ElementLinear

1 parent b666d17 commit 89c18d9Copy full SHA for 89c18d9

3 files changed

+384

-293

lines changed

dptb/nn
- embedding
  - lem.py
  - slem.py
- tensor_product.py

3 files changed

+384

-293

lines changed

Comments

(0)