- [ ] Encoder Model - [ ] Class Imbalance - [ ] Decoder Model - [ ] Add a callback that can record the models output to various prompts during training - [ ] Enable Early Stopping (Callback)