Logo
Insight Global

Machine Learning Engineer

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


Job Description

Insight Global is looking for a Machine Learning Engineer to join a client's newest AI team in San Jose, CA. 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. This role may also be based in Austin or Toronto and will require 10% travel onsite to customer facilities. Must Haves

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

Troubleshoot and alter the CUDA or ROCm layer of the software stack in support of large-scale AI partners. Run tests or simple models including LaMDA2 or Machine Learning models to ensure system stability and smooth operation of software stacks. Company Information

We aren’t just a staffing company. We’re a company that cares for others. Insight Global is dedicated to empowering people through the value of opportunity, built on genuine relationships and meaningful impact.

#J-18808-Ljbffr