Skip to content

metatensor-learn v0.2.2

Latest
Compare
Choose a tag to compare
@github-actions github-actions released this 16 May 14:29
· 20 commits to master since this release

This is a new backward-compatible release of metatensor-learn, with the following changes:

Added

  • Added torch-style activation function module maps to metatensor.learn.nn: ReLU, InvariantReLU, SiLU, and InvariantSiLU (#597)
  • Added torch-style neural network module maps to metatensor.learn.nn: LayerNorm, InvariantLayerNorm, EquivariantLinear, Sequential, Tanh, and InvariantTanh (#513)

Fixed

  • Set correct device for output of when torch default device is different than input device (#595)