retain_graph
* Adds global tracing flag * Removes retain_graph in favor of is_tracer
* add move and swap axis, and vmap for slice, concat, and gather