Stage Kv Cache Design For Efficient Transformer Deployment In Edge Systems-Saclay H/F - CEA

Le CEA est un acteur majeur de la recherche, au service des citoyens, de l'économie et de l'Etat.

Il apporte des solutions concrètes à leurs besoins dans quatre domaines principaux : transition énergétique, transition numérique, technologies pour la médecine du futur, défense et sécurité sur un socle de recherche fondamentale. Le CEA s'engage depuis plus de 75 ans au service de la souveraineté scientifique, technologique et industrielle de la France et de l'Europe pour un présent et un avenir mieux maîtrisés et plus sûrs.

Implanté au coeur des territoires équipés de très grandes infrastructures de recherche, le CEA dispose d'un large éventail de partenaires académiques et industriels en France, en Europe et à l'international.

Les 20 000 collaboratrices et collaborateurs du CEA partagent trois valeurs fondamentales :

- La conscience des responsabilités
- La coopération
- La curiosité

En tant que stagiaire au CEA, vous aurez l'opportunité de travailler au sein d'un environnement de recherche de renommée mondiale. Nos équipes sont composées d'experts passionnés et dédiés, offrant un cadre propice à l'apprentissage et à la collaboration. Vous aurez accès à des équipements de pointe et à des ressources de recherche de premier ordre pour mener à bien vos missions. Éventuellement, les travaux menés donneront lieu à une publication scientifique.

Contexte

Dans le cadre de ses activités sur de conception d'inférence légère pour l'embarqué, le LIAE s'intéresse à l'étude et la faisabilité de l'implémentation de ce type de fonctionnalité. L'idée principale est d'analyser et de quantifier les besoins et les limites pour un déploiement sur des architectures type CPU, GPU ou SoC...

Qu'attendons-nous de vous?

Dans ce contexte, l'objectif de ce stage est de :

· Fournir un état de l'art croisé et étayé à la fois des besoins de LLM choisis judicieusement et des différentes méthodes décrites dans la littérature,

· Développer et valider un modèle fonctionnel léger d'un « KV cache » en langage C++ dans l'outil du laboratoire,

· Mettre en oeuvre le mécanisme dans un modèle LLM retenu.

#Cea List

Profil :

Niveau demandé : 3ème année étude ingénieur ou master 2

Compétences : 1ère expérience avec Pytorch et C++, mise en oeuvre d'inférence, des notions sur les architectures matérielles serait un plus

Pièces à fournir : CV + Lettre de motivation + Relevés de notes des 3 dernières années