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.

Quick Facts

• Hybrid-by-design work mode and remote work opportunity within Poland
• Opportunity to work abroad for up to 60 days annually
• Business-driven relocation opportunities
• Open to working with contractors; B2B cooperation terms are agreed individually

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 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 and hands-on experience developing, training, and fine-tuning deep learning models
• Proven LLM experience
• Skills in designing, implementing, and iterating on neural network architectures for optimal performance across diverse tasks
• Ability to investigate and troubleshoot model performance issues, including gaps in compilation steps and kernels
• Experience designing, training, and deploying neural networks for various applications
• Solid understanding of machine learning fundamentals, including supervised and unsupervised learning techniques

Benefits

• Career development programs
• Thought leadership, mentoring, soft-skills, and well-being programs
• Certification opportunities: Anthropic, Gemini, GCP, Azure, AWS
• English classes
• Stable pay
• Employee Stock Purchase Plan participation with a 15% discount
• Benefits package including health insurance, multisport, and shopping vouchers
• Referral bonuses up to $2,000
• Offices with entertainment and relaxation zones, table tennis, football, free snacks, and coffee
• Corporate, social, and well-being events

Benefits listed above are available to employees only.

Only selected candidates will be contacted.