EnzymeAD / EnzymeAD/Enzyme-JAX
Decouple from JAX
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
Hi, is there any interest in pulling out the non-JAX specific stuff, such as autodiff of (Stable)HLO? It would reduce cognitive overhead for users, and simplify builds.
I frequently have trouble debugging build issues regarding JAX, and you're overriding the XLA commit used by JAX, so I'm not even sure I'm using anything from JAX.
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by identifying which parts of the project provide non-JAX functionality such as autodiff of StableHLO, and trace how the JAX and XLA dependencies enter the build. Done would mean a clearly scoped separation that reduces JAX-related build and debugging overhead without losing the non-JAX functionality.
Written by the indexing model from the issue text.
Assessment
- Domain
- compilers
- Issue type
- Refactor
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Quiet
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100