Lead ML&AI Engineer

We are looking for a Lead ML&AI Engineer to join a team building the next generation of computers for artificial intelligence. The company brings together experts in computer architecture, ASIC design, advanced systems and neural network compilers, with the team focused 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 workloadsDevelop and optimize the compiler and neural network stack supporting LLMs and CNNsCollaborate with experts in computer architecture, ASIC design and advanced systems to build next-generation computing solutionsDesign and implement AI/ML kernel operations tailored to GPU/AI-accelerator architecturesBuild and maintain ML models using PyTorch or TensorFlowEnsure kernel-level performance and correctness across modern neural network architectures such as transformersTake 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/AI5+ years of experience in C++ and CUDA/kernel programmingKnowledge in GPU/AI-accelerator architecturesExperience in building and working with ML models in PyTorch or TensorFlowUnderstanding of modern ML model architectures such as transformersEnglish 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 modelsSkills in designing, implementing and iterating on neural network architectures to achieve optimal performance on diverse tasksCapability to investigate and troubleshoot model performance issues including gaps in compilation steps and kernelsDemonstrated experience in designing, training and deploying neural networks for various applicationsSolid understanding of machine learning fundamentals including supervised and unsupervised learning techniques We offer We gather like-minded people:Top tech minds driving innovation in AI, cloud and digital platform modernizationSupportive team and agile, startup-like cultureHybrid by design mode and opportunity to work remotely within PolandChance to work abroad for up to 60 days annuallyBusiness-driven relocation opportunitiesWe provide growth opportunities:Career development programsThought leadership, mentoring, soft skills and well-being programsCertification (Anthropic, Gemini, GCP, Azure, AWS)English classesWe cover it all:Stable payParticipation in the Employee Stock Purchase Plan with a 15% discountBenefits package (health insurance, multisport, shopping vouchers)Referral bonuses up to $2,000Offices featuring entertainment and relaxation zones, table tennis and football, free snacks, coffee and moreCorporate, social and well-being eventsPlease, note:Benefits listed above are available to employees onlyWe are open for working with Contractors. Terms of B2B cooperation agreements are agreed individuallyWe will reach out to selected candidates exclusively EPAM is a leading global provider of digital platform engineering and development services. We are committed to having a positive impact on our customers, our employees, and our communities. We embrace a dynamic and inclusive culture. Here you will collaborate with multi-national teams, contribute to a myriad of innovative projects that deliver the most creative and cutting-edge solutions, and have an opportunity to continuously learn and grow. No matter where you are located, you will join a dedicated, creative, and diverse community that will help you discover your fullest potential.