Skip to content

2024-01-26

Latest
Latest
Compare
Choose a tag to compare
@mariogeiger mariogeiger released this 26 Jan 08:16
· 2 commits to main since this release
0.20.6
cc739f4

Added

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

Changed

  • replace jnp.ndarray by jax.Array
Assets 2