Please support JAX/Flax for the full release

#9
by HyperBlaze - opened

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?

Sign up or log in to comment