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
relax type annotations for some function arguments to also allow scalar float array inputs (instead of just floats) #15
Closed
copybara-service[bot] closed 7 months ago
relax type annotations for some function arguments to also allow scalar float array inputs (instead of just floats)