Machine Learning Engineer Intern
Core
Deploying state-of-the-art ML models (LLMs, CNNs) on custom AI hardware and optimizing performance for real workloads.
Role type
Machine Learning Engineer Intern (Systems/Hardware focus)
Builds
Custom AI hardware platforms running optimized ML models
Domain
AI hardware, RISC-V, compilers, semiconductors
Deliverable
production ML models
Required skills
Python, C++, CUDA, PyTorch, TensorFlow, kernel optimization, benchmarking
Preferred skills
RISC-V architecture, compiler flows, hardware-software co-design
Technologies
PyTorch, TensorFlow, CUDA, RISC-V
Responsibilities
Analyze ML model compilation and execution on Tenstorrent hardware; Improve kernels for computation and data movement; Run experiments to evaluate and improve performance; Support benchmarking and robustness testing
Seniority
Intern
Sourced via adzuna · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.
