Learning Python on your Android device is totally doable these days. Gone are the days when you needed a full computer setup.
add proper methods for numpy in all combinations with python / numpy types same for any combination of jax array as x and numpy/python as y-value (and the other way around) combination of python ...