AI memory architecture