consider value matrix shapes for Jax conversion#10

William-Baker wants to merge 2 commits into

google-deepmind:maingoogle-deepmind/tracr:mainfrom

William-Baker:Jax-conversion-ov-shape-considerationWilliam-Baker/tracr:Jax-conversion-ov-shape-considerationCopy head branch name to clipboard