3 ms·
If JAX had affine_grid() and grid_sample(), I'd be using it instead of PyTorch for my current project.
by eli_gottlieb 3y ago
If JAX had affine_grid() and grid_sample(), I'd be using it instead of PyTorch for my current project.
loading thread…