adding sgd_jit NOTE: This does NOT work