Lead AI Engineer with Machine Learning

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

Languages