AI-Hypercomputer / AI-Hypercomputer/JetStream
Remove jax dependencies in JetStream
Open
- Dominant language
- Python
- Stars
- 457
- Forks
- 67
- PR merge metrics
- No merged PRs in 30d
Description
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.
Contributor guide
Assessment
This issue has not been assessed yet.