jaxDecomp
jaxDecomp copied to clipboard
Fix slice unpad
Fixed slice_unpad, now it supports complex numbers
Ignore the formatting error. It is fixed in another PR Once you merge https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/17 and https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/12 and I wll rebase and this should be ok
This PR is good to go after the other two
@ASKabalan can you try to rerun your precommit locally to squash the formatting test error?
Yes all is green locally. But at some point this formatting error correction slipped here and I think it fixes this issue. Any ways this PR should come after the other two
rebased and fixed commit history