tfep.nn.dynamics

Dynamics layers for continuous normalizing flows.

Modules

egnn

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