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 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 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.
• Strong understanding of machine learning fundamentals, including supervised and unsupervised learning techniques.

We offer

• Hybrid-by-design work model and the opportunity to work remotely within Poland.
• Opportunity to work abroad for up to 60 days annually.
• Business-driven relocation opportunities.
• Career development programs.
• Thought leadership, mentoring, soft-skills, and well-being programs.
• Certification opportunities: Anthropic, Gemini, GCP, Azure, and AWS.
• English classes.
• 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 with entertainment and relaxation zones, table tennis, football, free snacks, coffee, and more.
• Corporate, social, and well-being events.

Please note

• Listed benefits are available to employees only.
• Contractors are welcome; B2B cooperation terms are agreed individually.
• Only selected candidates will be contacted.