Index

_

__call__() (PadGCN method) (PadGCNLayer method) (PadGCNPredicator method) (SparseGCN method) (SparseGCNLayer method) (SparseGCNPredicator method)

__init__() (EarlyStopping method) (PadGCN method) (PadGCNLayer method) (PadGCNPredicator method) (SparseGCN method) (SparseGCNLayer method) (SparseGCNPredicator method)

B

binary_cross_entropy_with_logits() (in module jaxchem.loss)

E

EarlyStopping (class in jaxchem.utils)

J

jaxchem.loss module jaxchem.models.gcn module

jaxchem.models.readout module jaxchem.utils module

M

module jaxchem.loss jaxchem.models.gcn jaxchem.models.readout jaxchem.utils

P

pad_graph_pooling() (in module jaxchem.models.readout) PadGCN (class in jaxchem.models.gcn)

PadGCNLayer (class in jaxchem.models.gcn) PadGCNPredicator (class in jaxchem.models.gcn)

S

sparse_graph_pooling() (in module jaxchem.models.readout) SparseGCN (class in jaxchem.models.gcn)

SparseGCNLayer (class in jaxchem.models.gcn) SparseGCNPredicator (class in jaxchem.models.gcn)

U

update() (EarlyStopping method)