AI layer splitting