API reference

Note

This is the documentation for metatensor-learn version 0.6.0. For other versions, look in the following pages:

Version 0.6

0.6.0

https://docs.metatensor.org//metatensor-learn-v0.6.0/learn/reference/index.html

Version 0.5

0.5.0

https://docs.metatensor.org//metatensor-learn-v0.5.0/learn/reference/index.html

Version 0.4

0.4.0

https://docs.metatensor.org//metatensor-learn-v0.4.0/learn/reference/index.html

Older versions

0.3.2

https://docs.metatensor.org//metatensor-learn-v0.3.2/learn/reference/index.html

0.3.1

https://docs.metatensor.org//metatensor-learn-v0.3.1/learn/reference/index.html

0.3.0

https://docs.metatensor.org//metatensor-learn-v0.3.0/learn/reference/index.html

0.2.3

https://docs.metatensor.org//metatensor-learn-v0.2.3/learn/reference/index.html

0.2.2

https://docs.metatensor.org//metatensor-learn-v0.2.2/learn/reference/index.html

0.2.1

https://docs.metatensor.org//metatensor-learn-v0.2.1/learn/reference/index.html

0.2.0

https://docs.metatensor.org//metatensor-learn-v0.2.0/learn/reference/index.html

0.1.0

https://docs.metatensor.org//metatensor-learn-v0.1.0/learn/reference/index.html

• Data utilites
  • Dataset
    • Dataset
    • IndexedDataset
  • Dataloader
    • DataLoader
  • Collating data
    • group()
    • group_and_join()
• Neural Network building blocks
  • Modules
    • Module
    • ModuleMap
    • Sequential
    • Linear
    • EquivariantLinear
    • Tanh
    • InvariantTanh
    • ReLU
    • InvariantReLU
    • SiLU
    • InvariantSiLU
    • LayerNorm
    • InvariantLayerNorm
    • EquivariantTransformation