jax

b995787c - Add "edge" support for pad (#2265)

Add "edge" support for pad (#2265) * Internal refactoring of jax.numpy.pad for greater readability. * Implement "edge" mode for pad * Remove unneeded comment per discussion