Retraining kwyk model (bayesian neural network) on kwyk dataset using nobrainer API

[hvgazula]

• [x] Try running the model until fit and document any issues
• [ ] Run with predict and see how the API needs to be adapted. This is because the kwyk model returns multiple outputs.