wip: checkpoint on layers