E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
87
stars
5
forks
source link
Add option to include mappings as weighting in mapped functions + rework unit tests to cover all variants + small fixes in docstrings #10
Closed
copybara-service[bot] closed 8 months ago
Add option to include mappings as weighting in mapped functions + rework unit tests to cover all variants + small fixes in docstrings