🚀 Feature request

Add better error message to HubertForCTC, Wav2Vec2ForCTC if labels are bigger than vocab size.

Motivation

Following this issue: huggingface/transformers#12264 it is clear that an error message should be thrown if any of the any of the labels are > self.config.vocab_size or else silent errors can sneak into the training script.

So w

Is there a way to train a bidirectional RNN (like LSTM or GRU) on trax nowadays?

After #1016 and #989, it would be nice to port ProdLDA example in Pyro to illustrate some usage of JAX nn frameworks flax/haiku in NumPyro. We might consider replacing SoftmaxNormal there with Dirichlet distribution. This is a good first issue for those who are interested in both NLP and Ba

Description

TIL that pip has a super nice Issue template page

that uses a [.github/ISSUE_TEMPLATE/config.yml](https://github.com/pypa/pip/blob/e6414d6db6db37951988f6f2b11ec530ed0b191d/.github/ISSUE_TEMPLATE/conf

See wikipedia table for example patterns: https://en.wikipedia.org/wiki/List_of_integrals_of_Gaussian_functions