iree-org/iree-jax issues and pull requests

#82 - Add CONTRIBUTING.md

Pull Request - State: closed - Opened by marbre 7 months ago

#81 - Add experimental code for custom attention op

Pull Request - State: open - Opened by NatashaKnk 9 months ago

#80 - Use stable URL for IREE pip-release-links.

Pull Request - State: closed - Opened by ScottTodd 10 months ago - 1 comment

#79 - Remove dependencies on old `iree-tools-xla` package

Issue - State: open - Opened by ScottTodd about 1 year ago

#78 - Change away from deprecated `iree` python backend

Pull Request - State: open - Opened by rsuderman over 1 year ago

#77 - "RuntimeError: Unknown backend iree"

Issue - State: open - Opened by WoongQ over 1 year ago - 2 comments

#76 - minor doc fix README.md

Pull Request - State: open - Opened by spundun over 1 year ago

#75 - Update IREE to 20230710.578

Pull Request - State: closed - Opened by jpienaar over 1 year ago

#74 - Update input type

Pull Request - State: closed - Opened by jpienaar over 1 year ago

#73 - Flip to new form

Pull Request - State: closed - Opened by jpienaar over 1 year ago

#72 - iree-jax fail when exporting bfloat16 parameters

Issue - State: open - Opened by wangkuiyi over 1 year ago

#71 - Remove reference to `IREE_BUILD_LEGACY_JAX`.

Pull Request - State: closed - Opened by ScottTodd almost 2 years ago

#70 - How to enable iree-compiler, iree-runtime as iree building from source.

Issue - State: closed - Opened by redbopo almost 2 years ago - 6 comments

#69 - Add support for passing `ShapeDtypeStruct` to `iree.jax.like`

Pull Request - State: closed - Opened by phoenix-meadowlark almost 2 years ago
Labels: enhancement

#68 - Make config.py confgurable from the command-line

Pull Request - State: closed - Opened by wangkuiyi almost 2 years ago

#67 - Add -C - option to curl when downloading

Pull Request - State: closed - Opened by wangkuiyi almost 2 years ago

#66 - How to use a develop branch of IREE with iree-jax

Issue - State: closed - Opened by wangkuiyi almost 2 years ago - 1 comment

#66 - How to use a develop branch of IREE with iree-jax

Issue - State: closed - Opened by wangkuiyi almost 2 years ago - 1 comment

#65 - Added a notification of the gpt-2 test to discord notifications

Pull Request - State: closed - Opened by rsuderman almost 2 years ago

#64 - Fix tests and bumping scrips

Pull Request - State: closed - Opened by rsuderman almost 2 years ago

#63 - Some gpt2 fixes

Pull Request - State: closed - Opened by rsuderman almost 2 years ago

#62 - Update versions for jax and iree

Pull Request - State: closed - Opened by rsuderman almost 2 years ago

#61 - Update test_head_and_bump.yaml

Pull Request - State: closed - Opened by jpienaar almost 2 years ago - 1 comment

#60 - JAX tracing op

Issue - State: open - Opened by jpienaar almost 2 years ago

#59 - Add MLIR function finetune to GPT-2 export.py

Pull Request - State: closed - Opened by wangkuiyi almost 2 years ago

#58 - Finetune GPT-2

Pull Request - State: closed - Opened by wangkuiyi almost 2 years ago

#57 - Dynamic_api_test in jax failing when returning dynamic shape

Issue - State: closed - Opened by oliverdutton almost 2 years ago - 9 comments

#56 - Change batch size to 1 and add an example

Pull Request - State: closed - Opened by wangkuiyi almost 2 years ago

#55 - [JAX] Replace uses of jax.xla_computation() with jax.jit().lower().

Pull Request - State: closed - Opened by jpienaar about 2 years ago

#54 - models/gpt2/test_jax.py failed

Issue - State: closed - Opened by wangkuiyi about 2 years ago - 2 comments

#53 - IR value tracing of Mapping and Iterable interfaces

Pull Request - State: closed - Opened by sogartar about 2 years ago - 8 comments

#52 - Format python code

Pull Request - State: open - Opened by sogartar about 2 years ago

#51 - Add a tutorial

Pull Request - State: closed - Opened by wangkuiyi about 2 years ago - 3 comments

#50 - The minimalist Program-derived class

Issue - State: closed - Opened by wangkuiyi about 2 years ago - 1 comment

#49 - models/gpt2/test_jax.py failed

Issue - State: open - Opened by wangkuiyi about 2 years ago

#48 - Update to jax/jaxlib 0.4.2

Pull Request - State: closed - Opened by rsuderman about 2 years ago

#47 - How to install iree-jax

Issue - State: closed - Opened by wangkuiyi about 2 years ago - 4 comments

#46 - python setup.py develop errors

Issue - State: closed - Opened by wangkuiyi about 2 years ago - 1 comment

#45 - Make the example in README.md runnable

Pull Request - State: closed - Opened by wangkuiyi about 2 years ago - 4 comments

#44 - Installing iree-jax on M1 Mac errors

Issue - State: closed - Opened by wangkuiyi about 2 years ago - 3 comments

#43 - Add a test for training MNIST on CUDA and CPU that verifies same result as Jax

Pull Request - State: open - Opened by sogartar about 2 years ago - 1 comment

#42 - Add github job for checking code formatting

Pull Request - State: open - Opened by sogartar about 2 years ago - 2 comments

#41 - Add posting to discord hook

Pull Request - State: closed - Opened by jpienaar about 2 years ago

#40 - Fix for Jax/Jaxlib 0.4.1 integration

Pull Request - State: closed - Opened by rsuderman about 2 years ago

#39 - Update IREE to 20221121.334

Pull Request - State: closed - Opened by rsuderman about 2 years ago

#38 - Update iree-jax to generate stablehlo instead of mhlo IR

Pull Request - State: closed - Opened by not-jenni about 2 years ago

#37 - Update IREE to 20221003.285

Pull Request - State: closed - Opened by jpienaar over 2 years ago

#36 - Make tests accept both generic and custom form ml_program

Pull Request - State: closed - Opened by jpienaar over 2 years ago

#35 - Enable unregistered dialects temporarily

Pull Request - State: closed - Opened by jpienaar over 2 years ago

#34 - Using nightly versions of jaxlib

Issue - State: open - Opened by jpienaar over 2 years ago