If you want to contribute to Mini-RETURNN it is recommended to first open an issue to announce your plans. Please do not forget to apply "black" code formatting, and try to follow existing code structure.