Role overview
We are looking for an Image Processing / ML Engineer to join a high-tech project in the field of computer graphics and visual computing.
The project is focused on image processing and machine learning, including modern neural network architectures such as CNNs, transformers, and diffusion models.
The work is closely connected with GPU technologies and high-performance computing.
You will work with experienced engineers on advanced solutions for image analysis, rendering, and ML pipelines.
What you'll work on
Develop and improve image processing and machine learning solutions
Work with neural networks (CNN, transformer, diffusion models)
Implement and optimize ML models using GPU acceleration
Analyze performance and improve efficiency of algorithms
Collaborate with engineers from different technical domains
Read and understand technical documentation
Participate in design discussions and technical reviews
What we're looking for
Experience around image processing
ML background in CNN, transformer, and diffusion
Languages/API: Python, Pytorch, Cuda
C++, HLSL, D3D12
Ability to work with APIs and technical documentation
3d graphics knowledge a big advantage (but not required)
GPU HW knowledge a big advantage