Lead ML&AI Engineer
E
EPAM SystemsArtificial Intelligence
Opportunity to work remotely within PolandFull-TimeLead
Salary not disclosed
Apply NowOpens the employer's application page
Job Details
- Languages
- English (B2 level or higher)
- Experience
- 5+ years
- Required Skills
- Machine LearningPyTorchC++Tensorflow
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 in 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
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
View Full Description & ApplyYou'll be redirected to the employer's site
