Published onMay 8, 2021Neural style transfer in the JAX ecosystempythonjaxhaikudeep-learninggpu-computingstyle-transferA small hobby project about style transfer in JAX using neural networks that I have been working on for a couple of weeks.