C++/Machine Learning Engineer, AI Models Training

Job Description

Tenstorrent is seeking a C++ Machine Learning Engineer to join their team in Warsaw or Gdansk, Poland. The ideal candidate will play a crucial role in the development of ML training framework for development and training ML models. The candidate will have expertise in Python and PyTorch and will be responsible for designing, training, and fine-tuning complex models that will be applied to solve real-world challenges. Tenstorrent values collaboration, curiosity, and a commitment to solving hard problems.

Responsibilities:

• Collaborate with cross-functional teams to understand project requirements and translate them into effective AI solutions.
• Work on machine learning framework.
• Add support for new layers.
• 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.
• Understand how models map onto the Tenstorrent devices through compilation steps and kernels, investigate any gaps in functionality and performance, propose innovative solutions to mitigate any issues.
• Stay current with the latest research and developments in the field of machine learning and AI, applying relevant advancements to our projects.
• Contribute to the design and implementation of scalable and efficient AI pipelines.
• Work closely with engineers to integrate AI models into production systems.

Requirements:

• Bachelor's or higher degree in Computer Science, Engineering, or related field (or equivalent practical experience).
• Strong C++ and low level programming expertise.
• Linear algebra knowledge.
• Hands-on experience with PyTorch for developing and training deep learning models is a big plus.
• Demonstrated experience in designing, training, and deploying neural networks for various applications.
• Solid understanding of machine learning fundamentals, including supervised and unsupervised learning techniques.
• Excellent problem-solving skills and the ability to approach challenges with creativity and a strong analytical mindset.
• Familiarity with debugging techniques and a knack for identifying issues and finding effective solutions.
• Experience with C++ and kernel programming, knowledge in GPU/AI-accelerator architectures are pluses but not mandatory.
• A self-motivated individual with a strong desire to learn, explore, and excel in a fast-paced, innovative environment.
• Strong communication skills and the ability to work collaboratively in a team setting.

Tenstorrent offers:

• A highly competitive compensation package and benefits.