Deep Learning Libraries Engineer

1 Year ago • 3 Years + • Research Development

Job Summary

Job Description

Rivos is seeking a Deep Learning Libraries engineer to develop optimized Deep Learning operators for their SIMT machine. The role involves contributing to the development and optimization of operators for both training and inference of Deep Neural Networks. This position offers the opportunity to influence hardware architectural decisions for improved performance and power efficiency. The engineer will also contribute to client software, compiler, runtime, and simulators as part of a vertical development approach to define future solutions. Responsibilities include designing and implementing critical parts of DL operator libraries and kernels for PyTorch, contributing to performance analysis and simulators, and collaborating with silicon design and architecture teams.
Must have:
  • 3+ years software library development (C, C++)
  • Strong C++ programming skills
  • Strong parallel programming knowledge
  • Strong computer architecture background
  • Strong deep learning background
  • Excellent problem-solving
  • Excellent communication skills
  • Highly self-motivated
  • Ability to work in a team
Good to have:
  • Experience with PyTorch

Job Details

Rivos is developing optimized Deep Learning operators for its SIMT (Single Instruction Multiple Threads) machine, providing optimal use of the compute units the HW exposes. You will contribute to development and optimization of many operators used for both training and inference of Deep Neural Networks. In this process you will be able to influence the architectural decision of the HW engine to deliver more performant and more power efficient solutions. In a vertical development approach you will be contributing extensively to all the other parts of the solution: client software, compiler, runtime, simulator to help define the next generations of our solution.

Responsibilities

    • As a Deep Learning Libraries engineer, you will own or participate in the following
    • design and implement critical parts of the DL operators libraries, including kernels used by PyTorch
    • contribute to the performance analysis flow to guide optimization work
    • contribute to the functional and performance ISA simulators
    • collaborate cross-functionally with Silicon design, architecture experts, and other teams across the company

Requirements

    • at least 3 years of experience in software library development (C, C++)
    • strong C++ programming skills
    • strong knowledge of parallel programming languages
    • strong background in computer architecture and deep learning
    • experience with PyTorch a plus
    • excellent skills in problem solving, written and verbal communication, excellent organization skills, and highly self-motivated.
    • ability to work well in a team and be productive under aggressive schedules
Education and Experience
PhD, Master’s Degree or Bachelor’s Degree in technical subject area.

Similar Jobs

Varonis  - Manager of Customer Success

Varonis

United States (Remote)
4 Months ago
Coda - Business Development Manager

Coda

Los Angeles, California, United States (Hybrid)
4 Months ago
Taboola - Solution Engineer

Taboola

New Delhi, Delhi, India (On-Site)
10 Months ago
blue isle studios - UE5 Gameplay Programmer

blue isle studios

(Remote)
1 Year ago
Workato - Enterprise Development Representative

Workato

Denver, Colorado, United States (On-Site)
2 Weeks ago
Toast - Director of Software Engineering - AI and Data

Toast

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
bytedance - Research Scientist Graduate (eCommerce Recommendation)

bytedance

Seattle, Washington, United States (On-Site)
1 Month ago
Ansys - R&D Engineer II

Ansys

Evanston, Illinois, United States (Hybrid)
2 Weeks ago
Visa - Senior Director – AI, Data and Optimization

Visa

Atlanta, Georgia, United States (Hybrid)
1 Week ago
Apple - Computational Support for Machine Learning and Transformers

Apple

Cupertino, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Tesla - Delivery Advisor

Tesla

London, England, United Kingdom (On-Site)
5 Months ago
Playtika - VIP Account Management Team Leader

Playtika

Romania (Hybrid)
9 Months ago
Linden lab - Senior Growth Marketing Manager

Linden lab

San Francisco, California, United States (Hybrid)
3 Months ago
Brillio - Data Scientist

Brillio

Bengaluru, Karnataka, India (Hybrid)
10 Months ago
WebFX - Jr. Digital Public Relations Specialist

WebFX

Harrisburg, Pennsylvania, United States (On-Site)
9 Months ago
Any Desk - Sales Support Specialist

Any Desk

Clearwater, Florida, United States (Hybrid)
2 Weeks ago
Minted - Financial Analyst

Minted

(Remote)
3 Months ago
Nium - Manager - Revenue Operations Lead ( FP&A)

Nium

Mumbai, Maharashtra, India (Hybrid)
1 Month ago
Ingenuity Studios, LLC - Nuke Compositor

Ingenuity Studios, LLC

Vancouver, British Columbia, Canada (Remote)
1 Year ago
Saronic Technologies - Staff Forward Deployed Engineer

Saronic Technologies

Austin, Texas, United States (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Nintendo - Bilingual Project Manager (Japanese)

Nintendo

Redmond, Washington, United States (Hybrid)
10 Months ago
Zelis  - FP&A Manager, Cashflow and Capex

Zelis

St. Louis, Missouri, United States (Remote)
1 Week ago
crate entertainment  - Senior 3D Artist

crate entertainment

United States (Remote)
3 Months ago
beghou consulting - Consultant - Commercial Operations & Analytics

beghou consulting

Emeryville, California, United States (Hybrid)
2 Months ago
rivos - Senior Member of Technical Staff

rivos

Austin, Texas, United States (Hybrid)
2 Months ago
Medsender - Senior Software Engineer

Medsender

New York, United States (Hybrid)
6 Months ago
WebTech Corporation - CNC Operator

WebTech Corporation

Duncan, South Carolina, United States (On-Site)
2 Weeks ago
Adtran - Treasury Analyst

Adtran

Huntsville, Alabama, United States (On-Site)
3 Months ago
DraftKings - HR Systems Analyst (Time & Attendance)

DraftKings

United States (Remote)
1 Week ago
Apple - Manager, Systems Engineering

Apple

Austin, Texas, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Apple - AIML - Staff Machine Learning Engineer, Answers, Knowledge & Information

Apple

Seattle, Washington, United States (On-Site)
2 Weeks ago
Apple - Machine Learning Engineer, GenAI & LLM - AiDP - IS&T

Apple

Sunnyvale, California, United States (On-Site)
2 Months ago
Ansys - R&D Engineer II

Ansys

Chengdu, Sichuan, China (On-Site)
2 Weeks ago
Beta Craft - Sr. Python AI/ML Developer

Beta Craft

Pune, Maharashtra, India (On-Site)
4 Months ago
Mendix - Learning Developer – AI Focus

Mendix

Rotterdam, South Holland, Netherlands (Hybrid)
2 Weeks ago
Krafton - Lead of Physical AI Agent, Research Scientist

Krafton

Seoul, South Korea (On-Site)
3 Months ago
broadcom - Principal R&D Engineer, Software

broadcom

Palo Alto, California, United States (On-Site)
3 Weeks ago
zoox - Manager, Offline Driving ML

zoox

Foster City, California, United States (Hybrid)
6 Months ago
Nordson Corporation - Work student AI Software Engineer

Nordson Corporation

Eckental, Bavaria, Germany (On-Site)
1 Month ago
DevRev - Member of Applied AI Engineering Team ( ETL)

DevRev

Chennai, Tamil Nadu, India (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (On-Site)

Bengaluru, Karnataka, India (Hybrid)

Bengaluru, Karnataka, India (Hybrid)

Santa Clara, California, United States (Hybrid)

Hsinchu, Hsinchu City, Taiwan (Hybrid)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Hsinchu, Hsinchu City, Taiwan (Hybrid)

View All Jobs

Get notified when new jobs are added by rivos

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug