Adding jaxtyping and small shape improvements to inference - #293
Open
mattlevine22 wants to merge 2 commits into
Open
Adding jaxtyping and small shape improvements to inference#293mattlevine22 wants to merge 2 commits into
mattlevine22 wants to merge 2 commits into
