Representing local protein environments with atomistic foundation models