pytc.jastrow.nnΒΆ

Classes

EENMLP

MLP specifically for EEN with equivariant first layer.

JastrowDense

Dense layer for Jastrow factors.

MLP

Multi-layer perceptron network using Flax with residual connections.