Notebook Model implementations in jax/serket

Available models:

Type	Description	Link
	Train Vanilla Transformer for Arabic-English Machine Translation	Link
	Train PINN to solve Burger's Equation	Link
	Time Series Prediction using Bidirectional LSTM	Link
	Train a Convolutional LSTM for Video Prediction	Link
	Train Fourier Feature Network	Link
	Train ConvNet MNIST Classifier	Link
	Train UNet Image Segmenter	Link