API reference

Note

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

Version 0.2.2

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

Version 0.2.1

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

Version 0.2.0

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

Version 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
  • Modules
    • ModuleMap
    • Sequential
    • Linear
    • EquivariantLinear
    • Tanh
    • InvariantTanh
    • ReLU
    • InvariantReLU
    • SiLU
    • InvariantSiLU
    • LayerNorm
    • InvariantLayerNorm