Search Results for "compile"

...It supports reverse-mode differentiation (a.k.a. backpropagation) via grad as well as forward-mode differentiation, and the two can be composed arbitrarily to any order. What’s new is that JAX uses XLA to compile and run your NumPy programs on GPUs and TPUs. Compilation happens under the hood by default, with library calls getting just-in-time compiled and executed. But JAX also lets you just-in-time compile your own Python functions into XLA-optimized kernels using a one-function API, jit. Compilation and automatic differentiation can be composed arbitrarily, so you can express sophisticated algorithms and get maximal performance without leaving Python.

...The coverage spans undergraduate and early graduate topics, designed to show how different areas of math fit together—linear algebra, analysis, topology, number theory, and more—without going deeply into every subtopic. Because it is written in LaTeX (with supporting Asymptote or other tools), readers can compile their own version, and the repository integrates diagrams, flowcharts, and supplementary files.