tfep.nn.dynamics.egnn

E(n) equivariant graph neural network for continuous normalizing flows.

Classes

EGNNDynamics(node_types, r_cutoff[, ...])

E(n) equivariant graph neural network for continuous normalizing flows.