AI-Hypercomputer / AI-Hypercomputer/JetStream

Remove jax dependencies in JetStream

Aperta
#88 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
457
Fork
67
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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.

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.