User Guide
Reference
faenet
faenet.base_model
faenet.embedding
faenet.eval
faenet.fa_forward
faenet.force_decoder
faenet.frame_averaging
faenet.model
faenet.transforms
faenet.utils
__version__
BaseModel
PhysEmbedding
eval_model_symmetries()
reflect_graph()
rotate_graph()
transform_batch()
model_forward()
ForceDecoder
LambdaLayer
check_constraints()
compute_frames()
data_augmentation()
frame_averaging_2D()
frame_averaging_3D()
EmbeddingBlock
FAENet
InteractionBlock
OutputBlock
FrameAveraging
Transform
GaussianSmearing
RandomReflect
RandomRotate
base_preprocess()
get_pbc_distances()
pbc_preprocess()
swish()
FAENet: Frame Averaging Equivariant Network
Note
documentation in progress
Refer to faenet.model for the model definition or FAENet for the class definition.
Bye bye