differentiable_tmm
