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…
• 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