Add per-sample ODE horizons, batched RoPE/masking, Stiefel equivariant layer and enhanced ray marcher#2

jeevesh415 wants to merge 1 commit into

mainjeevesh415/HRM:mainfrom

codex/analyze-and-rectify-code-logic-flow-iyv1w4jeevesh415/HRM:codex/analyze-and-rectify-code-logic-flow-iyv1w4Copy head branch name to clipboard