What problems do you run into for maths with Python? For linear algebra and ML with Jax/Numpy I find it quite readable.
The way that types interact with methods _feels_ like maths. I.e. you have a method made up of mathematical operators, if you swap out an input, it typically does the natural and correct thing.