Spaces:

zero-gpu-explorers
/

README

Running

App Files Files Community

129

Does zero-gpu support JAX?

#112

by lnyan - opened Sep 22

Discussion

lnyan

ZeroGPU Explorers org Sep 22

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.

datasciguy

Sep 22

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.

julien-c

ZeroGPU Explorers org Sep 30

JAX is not currently supported as far as i know! cc @cbensimon

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