PyPie

PyPie statically catches tensor shape mismatches, using a dependent type checker seasoned with rank polymorphism. It's a Python DSL that can be embedded in any Python programs.