Reproduce “Handling JAX” on OpenResearch
Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.
Universal app for Apple silicon and Intel. Requires macOS 11 or later.
About this paper
Oghenevwogaga Ebresafe, Daniel Zheng, Gordon Plotkin, Ningning Xie
The authors present effecax, a JAX library that compiles algebraic effect handlers away for JAX’s first-order representation. It lets programmers express state and errors within standard loops, translating handlers before the backend. A Diffrax refactor uses about 22% fewer lines; cold-start forward compilation takes 1.86 s for Diffrax and 1.53 s with effecax. Across loops, only tail-resumptive and abortive handlers are supported.
Summary by alphaXiv.
Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.