2024-01-26

mariogeiger released this 26 Jan 08:16

· 49 commits to main since this release

cc739f4

Added

e3nn.where function
Add optional mask argument in e3nn.flax.BatchNorm

Changed

replace jnp.ndarray by jax.Array

Assets 2