Ai performance optimization

shape