Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Full-time - Entry Level

Little Rock, AR

5,001-10,000 employees

Web Search Portals, Libraries, Archives, and Other Information Services

About the position

The Software Engineer, Systems ML - Frameworks / Compilers / Kernels role at Meta involves working within the MTIA Software team to develop a comprehensive AI Compiler strategy that enhances the performance and deployment of deep learning and machine learning models. The position focuses on optimizing AI frameworks, compiler stacks, and high-performance kernels for specialized hardware architectures, contributing to the advancement of the PyTorch AI framework and collaborating with AI researchers and hardware design teams.

Responsibilities

• Development of software stack focusing on AI frameworks, compiler stack, and high-performance kernel development.
• Contribute to the development of PyTorch AI framework core compilers for new AI hardware accelerators.
• Analyze deep learning networks and implement compiler optimization algorithms.
• Collaborate with AI research scientists to enhance deep learning models in various domains such as Recommendation systems and NLP.
• Performance tuning and optimizations of deep learning framework and software components.

Requirements

• Proven C/C++ programming skills.
• Currently pursuing or has obtained a Bachelor's degree in Computer Science, Computer Engineering, or a relevant technical field.
• Experience in AI framework development or accelerating deep learning models on hardware architectures.

Nice-to-haves

• A Bachelor's degree in Computer Science or related field with 4+ years of experience in AI framework development, or a Master's degree with 2+ years, or a PhD in a relevant field.
• Knowledge of GPU, CPU, or AI hardware accelerator architectures.
• Experience with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT.
• Experience with CUDA programming, OpenMP/OpenCL programming, or AI hardware accelerator kernel programming.
• Experience with compiler optimizations such as loop optimizations and vectorization.

Benefits

• Competitive hourly wage and annual salary with bonus and equity options.
• Comprehensive health benefits including medical, dental, and vision insurance.
• 401(k) retirement savings plan with company matching.
• Paid time off and holidays.
• Opportunities for professional development and continued education.