Image Processing / ML Engineer

Project description

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.

Responsibilities

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

Skills

Must have

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

Nice to have

3d graphics knowledge a big advantage (but not required)

GPU HW knowledge a big advantage