Note: The job is a remote job and is open to candidates in USA. EPAM Systems is seeking a Lead AI Engineer to join a cutting-edge team building computers specifically for AI. The role involves designing and implementing neural network architectures, training machine learning models, and collaborating with cross-functional teams to enhance AI solutions.
Responsibilities
- Design, implement, and iterate on neural network architectures using PyTorch to achieve optimal performance on diverse tasks
- Train, validate, and fine-tune machine learning models using relevant datasets to ensure high accuracy and robustness
- Investigate and troubleshoot model performance issues, iterating on models and techniques to continuously improve results
- Analyze how models map onto Tenstorrent devices through compilation steps and kernels
- Identify gaps in functionality and performance, proposing innovative solutions to mitigate any issues
- Collaborate with cross-functional teams working on compilers, neural networks, LLMs, and CNNs
- Deploy neural networks for various applications, ensuring scalability and reliability
Skills
- 5+ years of experience in machine learning and AI engineering
- At least 1 year of relevant leadership experience
- Proficiency in Python programming with hands-on experience in PyTorch for developing and training deep learning models
- Expertise in designing, training, and deploying neural networks for various applications
- Understanding of machine learning fundamentals, including supervised and unsupervised learning techniques
- Background in investigating and troubleshooting model performance issues
- Capability to map models onto specialized AI hardware through compilation steps and kernels
- Strong English communication skills (B2 level or higher)
- Experience with C++ and kernel programming
- Knowledge of GPU/AI-accelerator architectures
Benefits
- Diversity of assignments and projects
- Personal development plan
- Mentoring programs and leadership development
- Certification and professional development support
- Access to learning platforms including more than 2,500 internal courses and the LinkedIn Learning library with 20,000+ courses
- English courses taught by certified teachers
- Extra leave days
- Referral bonuses
- Private health insurance
- Recreation office zones with tea, coffee and snacks
- Sports and game consoles
- IT equipment and Microsoft's Software Assurance Home Use Program (HUP)
Company Overview
Company H1B Sponsorship