JAXArrayContext - Githubissues

inducer / arraycontext

Choose your favorite numpy-workalike!

6 stars 11 forks source link

Closed kaushikcfd closed 2 years ago

kaushikcfd commented 3 years ago

TODO:

~~[ ] Map loopy IR to python callable consisting of non-stateful array operations.~~
- Dropping because translating logic is complicated, most likely unnecessary.
[x] Test JAXArrayContext
[x] Needs https://github.com/inducer/pytato/pull/233

kaushikcfd commented 2 years ago

alexfikl, inducer: Thanks a lot for the suggestions!