Workloads Engineer

Workloads Engineer

(AI Systems / HW-SW Optimization)

Role Overview

This is not a traditional software engineering role.

We are looking for a Workloads Engineer responsible for translating AI models into efficient, production-ready execution on a new hardware + software stack. The role sits at the intersection of AI model understanding, systems engineering, and low-level performance optimization .

You will work across the full stack — from AI model structure down to hardware execution, ensuring that workloads are efficient, scalable, accurate, and robust on next-generation compute platforms.

Key Responsibilities

Analyze AI model architectures (including LLMs) and translate them into optimized execution workloads for custom HW/SW platforms

Design and implement high-performance software components for AI frameworks and runtime environments

Optimize AI workloads for:

performance (latency / throughput)

memory efficiency

parallel execution

numerical accuracy and stability

Identify and remove performance bottlenecks across the stack (model → runtime → hardware)

Contribute to design decisions for AI execution stack and system architecture

Support deployment and scaling of AI workloads in real-world environments

Required Qualifications

Bachelor’s or Master’s degree in Computer Science, Mathematics, Engineering, or related field

5+ years of hands-on software engineering experience (or AI model development experience)

Strong programming skills in Python and C++

Strong algorithmic thinking and ability to solve complex computational problems

Solid understanding of AI model architectures, especially transformers and LLMs

Experience in performance optimization (compute, memory, and parallelization techniques)

Strong communication skills and ability to work in cross-functional teams

Nice to Have

Experience with AI frameworks such as PyTorch, JAX, TensorFlow (training or inference)

GPU programming experience ( CUDA, OpenCL ) or parallel computing systems

Experience with AI performance tuning (latency, throughput, memory footprint optimization)

Familiarity with distributed systems and model deployment pipelines

Understanding of computer architecture (CPUs, GPUs, accelerators, memory hierarchies)

Experience working close to hardware / compilers / runtime systems

What We Offer

Highly competitive salary, employment contract (Umowa o Pracę), and a comprehensive benefits package, including Medicover healthcare coverage.

Work on the performance-critical compute layer for next-generation AI accelerators

Direct impact on deep learning model efficiency and latency

Collaboration with experts in hardware, compilers, and systems

Challengi