Add features for interfacing with vectors / tensors

[ericpan64]

Problem

Formatting things outside of 2d for ML can be very confusing since math and programming intermingle (reshaping, applying specific transforms)

Requested feature

A cleaner interface for working with vectors and ML things (PyTorch, TensorFlow, JAX, Dask, etc.)

Not exactly sure how yet, though want to jot the idea down

Alternatives considered

... look into this deeply!

Additional context