Machine Learning Engineer

Minimum Requirements

• 3 - 5 years of Experience with CUDA or ROCm Programming
• Experience with PyTorch
• Experience with HIP
• Experience with Kubernetes
• Experience with TensorFlow

Responsibilities

• This individual will be responsible for troubleshooting and altering the CUDA or ROCm layer of the software stack in support of large-scale AI partners.
• They will work with our top partners and run tests or simple models including LaMDA2 or Machine Learning models to ensure the system is stable and the software stacks are running smoothly.

Job Overview

Insight Global is looking for a Machine Learning Engineer to join a client's newest AI team in San Jose, CA. This role can also be in Austin or Toronto, and will require 10% travel onsite to customer facilities. If this sounds like you, apply today!

Company information:
We aren’t just a staffing company. We’re a company that cares for others. Insight Global is a purpose-driven company that’s dedicated to empowering people through the value of opportunity.