Please support JAX/Flax for the full release

by HyperBlaze - opened May 23

May 23

I hope to see JAX and Flax implementation for the fully released model, as in the model card it explicitly says it was trained on them.
Will it be possible?

osanseviero

Google org May 26

Thanks! Yes, we'll do a full release integrating into popular OS tools

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment