One approach would be to write code first, then run it by AI to get a critique. I think that strikes a good balance between avoiding atrophy and still getting the benefits of the tool.