Add support for OpenEquivariance Kernels - #5

Merged

kleinhenz merged 8 commits into

mainfrom

openequivariance

Sep 6, 2025

Commits on Sep 5, 2025

Add fused conv.
ameya98
committed
2d80dcf
Add failing test for FusedConv.
ameya98
committed
bef1004
Tests are working.
ameya98
committed
7187493
Formatting.
ameya98
committed
bdf51da
Remove old ExperimentalTensorProduct
ameya98
committed
e307630

Commits on Sep 6, 2025

add openequivariance extra
kleinhenz
committed
1611e89
don't modify default dtype for all tests
kleinhenz
committed
4bd64ad
assert torch.allclose -> torch.testing.assert_close
kleinhenz
committed
5b5b8b7