AI

🧨 Stable Diffusion in JAX / Flax !

Hugging Face has released a version of the Stable Diffusion model optimized for use with the JAX and Flax frameworks. This allows developers to run the model on Google's TensorFlow or Apple's Turi Create, in addition to PyTorch. The optimization is based on research by the Hugging Face team.
Hugging Face has released a version of the Stable Diffusion model optimized for use with the JAX and Flax frameworks. This allows developers to run the model on Google's TensorFlow or Apple's Turi Create, in addition to PyTorch. The optimization is based on research by the Hugging Face team. --- Why it matters: This matters because it expands the range of tools that can be used with Stable Diffusion, potentially making it easier for researchers and developers to integrate into their workflows. Source: https://huggingface.co/blog/stable_diffusion_jax

This article was originally published at: https://huggingface.co/blog/stable_diffusion_jax