Senior C++ Software Engineer with CUDA/GPU/TPU

We are looking for a Senior C++ Software Engineer to join a team building the next generation of computers for artificial intelligence. The team focuses on the compiler and neural network stack, including LLMs and CNNs. This is a key developer role within the ML & AI team.

Responsibilities

• Write kernels in C++ for various operations used across AI and ML workloads.
• Develop and optimize the compiler and neural network stack supporting LLMs and CNNs.
• Collaborate with experts in computer architecture, ASIC design and advanced systems to build next-generation computing solutions.
• Design and implement AI/ML kernel operations tailored to GPU/AI-accelerator architectures.
• Build and maintain ML models using PyTorch or TensorFlow.
• Ensure kernel-level performance and correctness across modern neural network architectures such as transformers.
• Take ownership of key technical decisions as a lead developer within the ML & AI team.

Requirements

• 5+ years of experience in software engineering with a focus on ML/AI.
• 5+ years of experience in C++ and CUDA/kernel programming.
• Knowledge of GPU/AI-accelerator architectures.
• Experience in building and working with ML models in PyTorch or TensorFlow.
• Understanding of modern ML model architectures such as transformers.
• English proficiency at B2 level or higher.

Nice to have

• Proficiency in Python programming with hands-on experience in developing, training and fine-tuning deep learning models.
• Proven LLM experience.
• Skills in designing, implementing and iterating on neural network architectures to achieve optimal performance on diverse tasks.
• Capability to investigate and troubleshoot model performance issues, including gaps in compilation steps and kernels.
• Demonstrated experience in designing, training and deploying neural networks for various applications.
• Solid understanding of machine learning fundamentals, including supervised and unsupervised learning techniques.

We offer

• Top tech minds driving innovation in AI, cloud and digital platform modernization.
• Supportive team and agile, startup-like culture.
• Hybrid-by-design mode and opportunity to work remotely within Poland.
• Chance to work abroad for up to 60 days annually.
• Business-driven relocation opportunities.

Growth opportunities

• Career development programs.
• Thought leadership, mentoring, soft skills and well-being programs.
• Certification: Anthropic, Gemini, GCP, Azure and AWS.
• English classes.

Compensation and benefits

• Stable pay.
• Participation in the Employee Stock Purchase Plan with a 15% discount.
• Benefits package, including health insurance, multisport and shopping vouchers.
• Referral bonuses up to $2,000.
• Offices featuring entertainment and relaxation zones, table tennis and football, free snacks, coffee and more.
• Corporate, social and well-being events.

Benefits listed above are available to employees only. Contractors are welcome; terms of B2B cooperation agreements are agreed individually. Only selected candidates will be contacted.