desc.objectives.ForceBalance.jit

ForceBalance.jit()

Apply JIT to compute methods, or re-apply after updating self.