Inference › Drivers / Frameworks / Toolkits › PyTorch / JAX / Other Frameworks v
JAX for inference only on 4090

Thread Rating:
  • 0 Vote(s) - 0 Average
  • 1
  • 2
  • 3
  • 4
  • 5
Linear Mode
JAX for inference only on 4090
southden Offline
Junior Member
**
Posts: 26
Threads: 2
Joined: Aug 2026
Reputation: 0
#3
09-27-2026, 02:09 PM
Interesting, I hadn't tried that. I don't think your conclusion is far off. The biggest change for me was simplifying the stack instead of adding one more optimization layer.
still benchmarking everything
Find
Reply


Messages In This Thread
JAX for inference only on 4090 - by torchstation - 09-26-2026, 12:39 AM
Re: JAX for inference only on 4090 - by southden - 09-27-2026, 02:09 PM
Re: JAX for inference only on 4090 - by modgarden - 09-29-2026, 04:50 AM
Re: JAX for inference only on 4090 - by llmarch51 - 09-30-2026, 04:59 AM

Forum Jump:


Users browsing this thread: 1 Guest(s)