mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,129 Commits 504 Branches 263 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
Jake VanderPlas
81e627d5bd DOC: make API doc titles more uniform 2023-01-18 10:59:42 -08:00
Peter Hawkins
80aec7b25f Documentation improvements. 2022-03-08 09:37:33 -05:00
John Flynn
7fd7009c23 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.
 2020-09-18 13:30:58 -07:00
Peter Hawkins
b943b31b22
Add jax.image.resize. (#3703) 
* Add jax.image.resize.

This is a port of `tf.image.resize()` and the `ScaleAndTranslate` operator.

While I don't expect this implementation to be particularly fast, it is a useful generic implementation to which we can add optimized special cases as the need arises.
 2020-07-10 09:57:59 -04:00