Text this: Augmented neural ODEs