Replace function removed from jax api with equivalent#80

Merged

btalamini merged 1 commit intosandialabs:mainfrom btalamini:fix_jax_removed_row_stack

Feb 27, 2024

+2-2

Commits on Feb 27, 2024

Replace function removed from jax api with equivalent
btalamini
committed