Zaslat SMS: Augmented neural ODEs