IREE Infrastructure - Githubissues

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

254 stars 48 forks source link

IREE Infrastructure #41

Open mjsML opened 1 year ago

mjsML commented 1 year ago

Currently we are standing up IREE support for JAX.

Ideally we should wire on top:

[x] #42

Then wire the E2E tests:

[ ] #37
[ ] #38
[ ] #40
[x] #39
[ ] #15

mjsML commented 1 year ago

cc: @pjannaty @stellaraccident , please tag others who need to be tagged.

pjannaty commented 1 year ago

@mjsML thank you for the reminder. I was very close in getting the JAX+PJRT+IREE building and working when the PAXML work and OpenXLA summit happened. I'll try to carve out some time tonight.