I ran the regression example you suggest and the model is still not learning. I also ran my code with a small sklearn dataset as a sanity check about my setup and it is in fact learning for that simple case, so it’s definitely not something silly like that. I am able to mostly replicate the lack of learning if I randomize the target column in the small toy sklearn example, so I think I’m facing a deeper issue with my data generation process. Thank you for your time again.