Senior AI/ML Engineer with PyTorch, С++
Summary
Senior AI/ML Engineer designing, training, and deploying advanced deep learning models using PyTorch. Works on neural network architectures, model optimization, and deployment across various environments.
We are looking for a Senior AI/ML Engineer to join the ML & AI team, driving the design, training, and deployment of advanced deep learning models.
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, and iterate on models and techniques to continuously improve results
- Analyze how models map onto the client's devices through compilation steps and kernels
- Identify gaps in functionality and performance, and propose innovative solutions to mitigate any issues
- Deploy neural networks for various applications across different environments
- Apply machine learning fundamentals to ensure effective supervised and unsupervised learning outcomes
Requirements
- 3+ years of experience in Python programming, with hands-on expertise in PyTorch for developing and training deep learning models
- Proficiency in designing, training, and deploying neural networks for various applications
- Understanding of machine learning fundamentals, including supervised and unsupervised learning techniques
- Expertise in troubleshooting and optimizing model performance across diverse tasks and datasets
- English proficiency at B2 level or higher
Nice to have
- Familiarity with C++ and kernel programming
- Knowledge of GPU/AI-accelerator architectures
Benefits
CONTINUOUS UPSKILLING, LEARNING & DEVELOPMENT
- Diversity of tasks and projects
- Assessment center for objective review of competency level
- Personal development plan
- Mentoring programs and leadership development
- Certification and professional development support
- Access to learning platforms including more than 2,500 internal courses
- English courses taught by certified teachers
CORPORATE BENEFITS
- Extra leave days
- Referral bonuses
COMPENSATION PACKAGE
- Competitive compensation paid in USD
- Regular salary and performance reviews
MEDICAL & HEALTHCARE
- Private health insurance
- Well-being events
WORKING ENVIRONMENT
- Recreation areas and kitchens
- Tea, coffee and snacks
- Sports equipment and game consoles
- IT Equipment
- Microsoft’s Software Assurance Home Use Program (HUP)