ESM3 layer optimization