Can specify uniform params per layer