3 ms·

I know about dfdx in rust that has type checked tensors for rank and shape. Don't know if it ticks all your boxes