Tootfinder

JaxDecompiler: Redefining Gradient-Informed Software Design
Pierrick Pochelu
https://arxiv.org/abs/2403.10571 https://arxiv.org/pdf/2…

JaxDecompiler: Redefining Gradient-Informed Software Design
Among numerical libraries capable of computing gradient descent optimization, JAX stands out by offering more features, accelerated by an intermediate representation known as Jaxpr language. However, editing the Jaxpr code is not directly possible. This article introduces JaxDecompiler, a tool that transforms any JAX function into an editable Python code, especially useful for editing the JAX function generated by the gradient function. JaxDecompiler simplifies the processes of reverse engineer…