Beyond automatic differentiation – Google AI Blog