I played around with the idea of a language motivated by this same thought process last year: https://github.com/lukehoban/ten https://github.com/lukehoban/ten.

I played around with the idea of a language motivated by this same thought process last year: https://github.com/lukehoban/ten https://github.com/lukehoban/ten. * Succinct syntax and operators tailored to AI model definition * Fully statically typed tensors, including generic functions over tensor dimension and batch dimensions (...) * First-class hyper-parameters, model parameters and model arguments for explicit model specification * Einops-style reshaping and reductions - tensor dimensions are explicit not implicit