This was one of the reasons for me wanting to study the J programming language. I have a notion AI/ML programming might be better done using array languages.

Some books for studying Mathematics using J are listed here - https://code.jsoftware.com/wiki/Books