Job Description
Tenstorrent is seeking a Machine Learning Engineer to contribute to the development and deployment of advanced neural networks and machine learning models. This role is based in Warsaw or Gdansk, Poland, and offers the opportunity to work on cutting-edge AI technology. The ideal candidate will possess strong technical skills, a deep curiosity, and a passion for staying up-to-date with the latest advancements in the AI landscape.
Role involves:
- Collaborating with cross-functional teams to understand project requirements and translate them into effective AI solutions.
- Designing, implementing, and iterating on neural network architectures using PyTorch.
- Training, validating, and fine-tuning machine learning models using relevant datasets.
- Investigating and troubleshooting model performance issues.
- Understanding how models map onto the Tenstorrent devices through compilation steps and kernels.
- Staying current with the latest research and developments in the field of machine learning and AI.
- Contributing to the design and implementation of scalable and efficient AI pipelines.
- Working 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 proficiency in Python programming and hands-on experience with PyTorch.
- Demonstrated experience in designing, training, and deploying neural networks.
- Solid understanding of machine learning fundamentals.
- Excellent problem-solving skills and the ability to approach challenges with creativity and a strong analytical mindset.
- Familiarity with debugging techniques.
- Strong communication skills and the ability to work collaboratively in a team setting.
Tenstorrent offers:
- A highly competitive compensation package and benefits.