This one should be `@local_optimizer(WeightActs)`