Please support JAX/Flax for the full release

by HyperBlaze - opened 23 days ago

23 days ago

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 20 days ago

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