Inference
JAX for inference only on 4090 - Printable Version

+- Inference (https://www.datasculpting.ai/inference)
+-- Forum: Drivers / Frameworks / Toolkits (https://www.datasculpting.ai/inference/forumdisplay.php?fid=8)
+--- Forum: PyTorch / JAX / Other Frameworks (https://www.datasculpting.ai/inference/forumdisplay.php?fid=34)
+--- Thread: JAX for inference only on 4090 (/showthread.php?tid=494)

Pages: 1 2


Re: JAX for inference only on 4090 - embersmith - 09-30-2026

Need a little more detail here. Is this local-only or are you routing anything externally?


Re: JAX for inference only on 4090 - visionyard23 - 09-30-2026

Quote:Need a little more detail here. Is this local-only or are you routing anything externally?

My first guess would be this: On B200, the easiest win is often to back off the aggressive settings and see whether the problem survives. Not glamorous advice, but repeatable baselines save a lot of time.