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

• Challenging low-level performance engineering problems at the hardware-software boundary