jax

06d32808 - Inlined some Triton-specific abstractions

Inlined some Triton-specific abstractions * tensor is now just a container for dtype/shape with no extra methods; * constexpr is not used; * all APIs assume that the arguments are tensors and do no ->tensor conversion. PiperOrigin-RevId: 603894852