What problems do you run into for maths with Python? For linear algebra and ML with Jax/Numpy I find it quite readable.