JetStream icon indicating copy to clipboard operation
JetStream copied to clipboard

Published 20 hours ago • verified publisher icon google

Remove jax dependencies in JetStream

Open FanhaiLu1 opened this issue 9 months ago • 0 comments

There multiple jax code in JetStream, we should shift jax related code to engine implementation and remove jax dependencies in JetStream.

In the end, JetStream is orchestrator for Pytorch and Jax inference.

FanhaiLu1 avatar May 23 '24 22:05 FanhaiLu1