Der Artikel stellt JaxDecompiler vor, ein Werkzeug, das jede JAX-Funktion in äquivalenten Python-Code übersetzt. Dies ist besonders nützlich, um den von der Gradientenfunktion generierten JAX-Code zu bearbeiten.
JaxDecompiler besteht aus drei Hauptkomponenten:
Die Leistung des decompilierten Codes wurde in fünf Anwendungen getestet und ist mit der des ursprünglichen JAX-Codes vergleichbar. JaxDecompiler ermöglicht es Forschern und Entwicklern, den von JAX generierten Code besser zu verstehen, anzupassen und für die Interoperabilität zu nutzen.
다른 언어로
소스 콘텐츠 기반
arxiv.org
더 깊은 질문