Approximation par apprentissage d'opérations non linéaires pour des réseaux de neurones émergents

Description du poste

:

L’objectif du stage vise à considérer une implémentation entière tout en approchant au maximum des performances du réseau de référence. La mise en œuvre de méthode de quantification peut également être abordée de manière connexe.

Dans ce contexte, l’objectif de ce stage est d’identifier les fonctions non linéaires concernées dans les réseaux ViT/VisionMamba et de mettre en place les techniques d’apprentissage pour la génération des opérations non linéaires approximés dans un contexte embarqué. Les principales missions du candidat seront les suivantes :

· L’identification des opérations non-linéaires concernées,

· Le déploiement et validation de la méthode d’approximation par apprentissage par des tests simples,

· L’intégration des opérateurs dans des modèles ViT et Vision Mamba sur modèle de référence et quantifiés.

Moyens / Méthodes / Logiciels Linux/Python/Pytorch/ML Profil du candidat

Qu’attendons-nous de vous ?

Vous préparez un diplôme de niveau bac+5 dans le domaine des systèmes embarquées et du traitement de signal.

Vous êtes passionné par la recherche scientifique et technologique. Idéalement, vous possédez une première expérience de travail avec l’environnement Pytorch dans le cadre d’une mise en œuvre de réseau de neurones. Des notions sur les architectures matérielles sont un plus. La maitrise de lecture de papiers scientifiques de référence en anglais serait également appréciée.

Rejoignez-nous, venez développer vos compétences et en acquérir de nouvelles !

Vous avez encore un doute ? Nous vous proposons :

1. L'opportunité de travailler au sein d'une organisation de renommée mondiale dans le domaine de la recherche scientifique,
2. Un environnement unique dédié à des projets ambitieux au profit des grands enjeux sociétaux actuels,
3. Une expérience à la pointe de l’innovation, comportant un fort potentiel de développement industriel,
4. Des moyens expérimentaux exceptionnels et un encadrement de qualité,
5. De réelles opportunités de carrière à l’issue de votre stage
6. Une participation aux transports en commun à hauteur de 85%,
7. Un équilibre vie privée – vie professionnelle reconnu,
8. Un restaurant d'entreprise,
9. Une politique diversité et inclusion,

Conformément aux engagements pris par le CEA en faveur de l'intégration des personnes handicapées, cet emploi est ouvert à toutes et à tous. Le CEA propose des aménagements et/ou des possibilités d'organisation pour l'inclusion des travailleurs handicapés.