Is there an equivalent to the `Transformer` class of PyTorch in `transformers` in `flax`?
I am looking for an equivalent ofTransformerinflaxand I thought that maybe there would be one intransformersbut couldn’t find it.I haven’t seen a lot of “base models” except maybe resnets.