google-research / e3x

E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
https://e3x.readthedocs.io/stable/
Apache License 2.0
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

copybara-service[bot] commented 8 months ago

Add option to include mappings as weighting in mapped functions + rework unit tests to cover all variants + small fixes in docstrings