jax

22036372 - Rename master->main in JAX API and internals (#4178)

Rename master->main in JAX API and internals (#4178) * Started with #4174 * Renamed Trace.master to Trace.main * Renamed core.new_master and core.new_base_master