On line 59, there is a less-than-or-equals comparison between 0 and 1. Curious
https://github.com/meta-llama/llama3/blob/main/llama/model.p...
Is it `assert(0 <= (1 < ndim))` or `assert((0 <= 1) < ndim)`, or something even stranger like `assert(0 <= 1) < ndim`?
In linked code we can be confident that `0 <= 1`, so only `1 < ndim` should matter. In fact I'd expect peephole optimization to remove most of the code for `0 <= 1`