Machine Learning Engineer, AI Models
Santa Clara, California, United States
Full Time Senior-level / Expert USD 100K - 500K
Tenstorrent
At Tenstorrent, we build AI processors for machine learning workloads of today and tomorrow.Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. With AI redefining the computing paradigm, solutions must evolve to unify innovations in software models, compilers, platforms, networking, and semiconductors. Our diverse team of technologists have developed a high performance RISC-V CPU from scratch, and share a passion for AI and a deep desire to build the best AI platform possible. We value collaboration, curiosity, and a commitment to solving hard problems. We are growing our team and looking for contributors of all seniorities.
As a Machine Learning Engineer focused on AI models, you will play a pivotal role in the development and deployment of advanced neural networks and machine learning models. Your expertise in Python and PyTorch will be critical in designing, training, and fine-tuning complex models that will be applied to solve real-world challenges. The ideal candidate will not only possess strong technical skills but also exhibit a deep curiosity, investigative mindset, and a passion for staying up-to-date with the latest advancements in the AI landscape.
This role is hybrid, based out of Toronto, ON; Boston, MA; Santa Clara, CA; or Austin, TX.
Responsibilities:
- 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.
Experience & Qualifications:
- Bachelor's or higher degree in Computer Science, Engineering, or related field (or equivalent practical experience).
- Strong proficiency in Python programming and hands-on experience with PyTorch for developing and training deep learning models.
- Demonstrated experience in designing, training, and deploying neural networks for various applications.
- Solid understanding of machine learning fundamentals, including supervised and unsupervised learning techniques.
- Excellent problem-solving skills and the ability to approach challenges with creativity and a strong analytical mindset.
- Familiarity with debugging techniques and a knack for identifying issues and finding effective solutions.
- Experience with C++ and kernel programming, knowledge in GPU/AI-accelerator architectures are pluses but not mandatory.
- A self-motivated individual with a strong desire to learn, explore, and excel in a fast-paced, innovative environment.
- Strong communication skills and the ability to work collaboratively in a team setting.
Compensation for all engineers at Tenstorrent ranges from $100k - $500k including base and variable compensation targets. Experience, skills, education, background and location all impact the actual offer made.
Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.
Due to U.S. Export Control laws and regulations, Tenstorrent is required to ensure compliance with licensing regulations when transferring technology to nationals of certain countries that have been licensing conditions set by the U.S. government.
Our engineering positions and certain engineering support positions require access to information, systems, or technologies that are subject to U.S. Export Control laws and regulations, please note that citizenship/permanent residency, asylee and refugee information and/or documentation will be required and considered as Tenstorrent moves through the employment process.
If a U.S. export license is required, employment will not begin until a license with acceptable conditions is granted by the U.S. government. If a U.S. export license with acceptable conditions is not granted by the U.S. government, then the offer of employment will be rescinded.
Tags: Architecture Computer Science Deep Learning Engineering Excel GPU Machine Learning ML models Pipelines Python PyTorch Research Unsupervised Learning
Perks/benefits: Career development Competitive pay
More jobs like this
Explore more AI, ML, Data Science career opportunities
Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.
- Open Business Intelligence Engineer jobs
- Open Lead Data Analyst jobs
- Open Power BI Developer jobs
- Open Data Engineer II jobs
- Open Senior Business Intelligence Analyst jobs
- Open Data Science Manager jobs
- Open Marketing Data Analyst jobs
- Open MLOps Engineer jobs
- Open Junior Data Scientist jobs
- Open Data Scientist II jobs
- Open Business Intelligence Developer jobs
- Open Business Data Analyst jobs
- Open Product Data Analyst jobs
- Open Data Analytics Engineer jobs
- Open Data Analyst Intern jobs
- Open Sr Data Engineer jobs
- Open Principal Data Scientist jobs
- Open Sr. Data Scientist jobs
- Open Senior Data Architect jobs
- Open Data Engineering Manager jobs
- Open Junior Data Engineer jobs
- Open Big Data Engineer jobs
- Open Data Quality Analyst jobs
- Open Research Scientist jobs
- Open Azure Data Engineer jobs
- Open GCP-related jobs
- Open Java-related jobs
- Open Data quality-related jobs
- Open ML models-related jobs
- Open Business Intelligence-related jobs
- Open Data management-related jobs
- Open Privacy-related jobs
- Open PhD-related jobs
- Open Data visualization-related jobs
- Open Deep Learning-related jobs
- Open Finance-related jobs
- Open NLP-related jobs
- Open PyTorch-related jobs
- Open TensorFlow-related jobs
- Open LLMs-related jobs
- Open APIs-related jobs
- Open Generative AI-related jobs
- Open CI/CD-related jobs
- Open Snowflake-related jobs
- Open Consulting-related jobs
- Open Hadoop-related jobs
- Open Kubernetes-related jobs
- Open Data governance-related jobs
- Open Databricks-related jobs
- Open Airflow-related jobs