JAX offers several composable function transformations (jit, grad, vmap, etc.) that enable writing concise, accelerated code. Here we show how to add your own function transformations to the system, ...
Given the graph of a common function, (such as a simple polynomial, quadratic or trig function) you should be able to draw the graph of its related function. The graph of the related function can be ...