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 |