EnzymeAD / EnzymeAD/Enzyme-JAX
`all-reduce`s in sharded baroclinic model
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
XLA and MLIR modules for the for loop https://gist.github.com/giordano/9e680d1732dc56258da0e60c556e1eb9. This is running https://github.com/PRONTOLab/GB-25/pull/187 with https://github.com/EnzymeAD/Reactant.jl/pull/1171.
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reading the XLA and MLIR modules for the for loop in the linked gist, then compare the running GB-25 pull request #187 and Reactant.jl pull request #1171. Clarify the expected all-reduce behavior for the sharded baroclinic model; completion cannot be determined from the issue alone.
Written by the indexing model from the issue text.
Assessment
- Domain
- distributed-systems
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100