Improve performance on UM side

By pre-loading in memory the UM files when calling init_um, there is a performance improvement of 8x-17x (depends on the test)

Update Python wrapper