Klong/k looks ideal for this to have some practice with linear alg concepts:

http://t3x.org/klong/index.html

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