Spaces:
Running
Running
Does zero-gpu support JAX?
#112
by
lnyan
- opened
I am trying to deploy a Flax model with Zero GPU, but it seems that the initialized state disappears after Gradio starts.
Does Zero GPU support JAX? If not, are there any plans to add support? Thanks.
I'm completely new to Zero GPU, but the model I was using built with it was in fact a Flax model, so they must support it. Hopefully someone with more info will also comment.