Machine Learning Engineer

Job Description

ENGINEERING - UNREAL ENGINE
What We Do
Unreal-powered projects have been on the bleeding edge of real-time entertainment for over 20 years. Our team of engineering experts are always innovating to improve the tools and technology that empower content developers worldwide.
What You'll Do
We are looking for a Machine Learning Engineer to push the limits of our NeuralNetworkEngine (NNE). NNE is the backbone of many ML applications in and outside Epic and comprises multiple inference backends running on various target systems. You will work closely with the team and other teams across Epic to keep UE's ML inference engine at the forefront of the industry and thus you will be part of the foundational pillar for many ML products.
In this role, you will
• Improve the MLIR/LLVM compiler toolchain to produce more performant ML code
• Optimize neural network shader kernels within Unreal Engine
• Update, maintain and extend different inference backends inside NNE
• Collaborate with teams across Epic to integrate NNE into their products
What we're looking for
• Programming skills in C/C++ and Python
• Experience with GPU programming (CUDA, DirectX or Metal) and shader languages (HLSL or GLSL)
• Experience with ML frameworks such as PyTorch, TensorFlow, or JAX
• Experience optimizing ML models for real-time inference on GPUs
• Strong collaboration and communication skills
• Ability to design and train a neural network given a problem would be an asset
• Experience with MLIR and LLVM would be an asset

Note to Recruitment Agencies: Epic does not accept any unsolicited resumes or approaches from any unauthorized third party (including recruitment or placement agencies) (i.e., a third party with whom we do not have a negotiated and validly executed agreement). We will not pay any fees to any unauthorized third party. Further details on these matters can be found here.

Jobcode: Reference SBJ-wp4qop-216-73-216-177-42 in your application.