Logo
Insight Global

Machine Learning Engineer

Insight Global, San Jose, California, United States, 95199


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 OverviewInsight 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.

#J-18808-Ljbffr