Collaborate with cross-functional teams to understand project requirements and translate them into effective AI solutions.
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.
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.
Skills Required
Machine Learning
Python
Find more jobs at Tenstorrent
At Tenstorrent, we build AI processors for machine learning workloads of today and tomorrow.