Publier un stage
Détails de l'offre
Emploi > Stages > Informatique/Technologie > Chine > Shanghai > Détails de l'offre 

Deep Learning Performance Architect Intern

Entreprise non montrée
Chine  Shanghai, Chine
Stage, Informatique/Technologie, Anglais

Description du poste:

We are now looking for Compute/DL Architecture Performance Optimization Interns in our group! Are you passionate about exploring computer architectures for deep learning? Do you enjoy working at the intersection of hardware and software? (COMPANY NAME) is looking for world-class programmers and performance architects who like to continuously explore and mine the ultimate performance of each operator (or fusion operator) in deep learning networks, design and develop scalable modular infrastructure that can ship these highly optimized operators to different (COMPANY NAME) software libraries for training and inference.

What you'll be doing:
* Analyze the performance of various machine learning/DL algorithms on existing/new architectures
* Identify bottlenecks and propose creative solutions to improve them
* Develop high performance operators on (COMPANY NAME) GPUs for cuBLAS, TensorRT, cuDNN, cuSparse and cuTensor libraries
* Design and develop software for shipping and testing the GPU operators
* Build scalable automation for testing, integration, and release processes for publicly distributed deep learning libraries
* Configure, maintain, and build upon deployments of industry-standard tools (e.g., Kubernetes, Jenkins, Docker, CMake, Gitlab, Jira, etc)

What we need to see:
* Pursuing a B.S., M.S., or PhD degree in computer science (or similar)
* Strong programming skills in C/C++ development
* Familiar with GPU programming model and CUDA
* Good understanding about AI compilation technologies and experience with MLIR, TVM development
* Excellent problem solving skills, good communication and teamwork

(COMPANY NAME) is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you

Origine: Site web de l'entreprise
Publié: 22 Mai 2024
Type de poste: Stage
Secteur: Électronique grand public
Langues: Anglais
109.345 emplois et stages
dans 151 pays