To reply to your other question, we were working on a fresh Numerics/ShapedArray lib pulling pieces from S4TF that made sense without pulling in too much of the complexity, and building out from there. This may align with what you're trying to achieve. Let's discuss in slack.