The API looks fairly brittle, e.g. manually defining the loss and activation inside the pipeline instead of in the model itself. Have you or your customers used this in a large production environment?
Especially feedback like yours is super appreciated, as we can gauge opinions and evaluate them against our roadmap, so thanks :).