PyPie — The DSL for Tensors

1 min read Original article ↗

Safety: Statically catches type errors, like a tensor shape mismatch. Simplicity: Keeps programs high-level, with rank polymorphism. Compatibility: Runs on JAX and PyTorch.

Playground

Run an exmaple, or make a one of your own.