from probjax.nn.nets.autoregressive import AutoregressiveMLP
from probjax.nn.nets.coupling import CouplingMLP
from probjax.nn.nets.deepsets import DeepSet
from probjax.nn.nets.lru import LRU, LRUModel
from probjax.nn.nets.mlp import MLP
from probjax.nn.nets.transformer import Transformer
