PR #1795 Fix OOM when using PyTorch with JAX installed.

Fix OOM when using PyTorch with JAX installed. #1795

pcuenca merged 1 commit into main from fix-oom-jax

Don't initialize Jax on startup.

9b17c635

pcuenca requested a review from

williamberman 3 years ago

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

patil-suraj 3 years ago

patil-suraj approved these changes on 2022-12-21

pcuenca merged 60438389 into main 3 years ago

pcuenca deleted the fix-oom-jax branch 3 years ago

Reviewers

patil-suraj

williamberman

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone