Add an assertion layer

[pranavm-nvidia]

In cases where we have composite APIs comprised of other APIs, we can have errors from some intermediate operation. In many cases, we can preempt these errors by inserting checks at the beginning, which can potentially allow for better error messages. However, we need to do so at runtime so that we have access to, e.g., runtime shapes. Consequently, we need an assertion layer that we can insert into the compute graph.

We can use this to enhance the tp.stack error messages in cases where the input shapes are not compatible.