Utility API

This is the API for the utils module of metatrain.

• Additive models
• Data
• Architectures
• Augmentation
• Device
• Dtype
• Errors
• Evaluating a model
• External Naming
• IO
• Jsonschema
• Logging
• Long-range
• Loss
• Metrics
• Neighbor lists
• Custom omegaconf functions
• Output gradient
• Averaging predictions per atom
• Scaler
• Summing over atoms
• Data type and device transfers
• Unit handling