jax

7fd7009c - Expose scale_and_translate as a public function, fix a bug in implementation when translation is not 0.

Expose scale_and_translate as a public function, fix a bug in implementation when translation is not 0. Change implementation to use native JAX everywhere allowing vmaping and gradients wrt scale and translation.