Software Engineer-AI Model Bring Up
Salary not listed • Tokyo, Japan • Full-time
Posted 11mo ago
This job has been open for more than 60 days. It may no longer be actively hiring, so check with the company before putting a lot into an application.
Remote Work Policy
On-site
Employment Type
Full-time
Categories
LLM Engineer
About the job
This role sits at the intersection of technical expertise and customer engagement, focused on helping customers and internal teams bring up and optimize AI models on Tenstorrent platforms. As a Software Engineer on the AI Models / System Bring-Up team, you will be responsible for porting, validating, and optimizing AI models on Tenstorrent platforms. Working across models, runtime software, and hardware, you will help connect new models, models not yet running on Tenstorrent hardware, and research-stage workloads to stable, high-performance systems.
Responsibilities
- Port, optimize, and validate AI models such as LLMs, CNNs, recommendation models, and vision models on Tenstorrent hardware.
- Port and integrate models into Tenstorrent toolchains and runtime environments.
- Design and run experiments to evaluate model accuracy, performance, and stability.
- Debug cross-stack issues and collaborate with cross-functional teams.
Requirements
- Hands-on experience with deep learning models in at least one major framework such as PyTorch, TensorFlow, or JAX.
- Strong Python or C++ skills with a solid understanding of neural network architectures, training, and inference fundamentals.
- Comfortable developing in Linux environments and debugging issues across software, runtime, and hardware layers.
- Collaborative and curious, with a degree in Computer Science, Engineering, Applied Mathematics, or a related field, or equivalent practical experience.
- Business-level Japanese and sufficient English proficiency.