Posted:1 month ago| Platform: Foundit logo

Apply

Work Mode

On-site

Job Description

MS/BS degree in Computer Science or an equivalent

Deep Knowledge of C/C++ and Python programming

Experience with Linux Commands is must

Experience withScripting language like bash/powershell

Understanding of various python ML frameworks like Pytorch, Transformers etc

Understanding of variouslanguage and compiler for writing highly efficient custom Deep-Learning GPU Kernels.like Triton/Jax

Hands on DebuggingExperience with gdb, valgrind etc

Experience andunderstanding ofAI Models and Inferencing Engines like vllm/ollama/llama.cpp/sglang

Experience with Profiling tools needed to debug CUDA/ROCm Kernels like nsys/rocprof is a plus.

Knowledge of GPU architecture,PC architecture

Experience in writing ROCM/CUDA Kernels/Shader

Deep understanding and experience in implementation of Machine learning and AI algorithm.

Good communication skills and able to work with stakeholders effectively

Knowledge of x86 assembly language and x86/x64 CPU instructions is a plus.

Responsibilities:

1.Work on latest machine learning technologies

2.Work on supporting for latest Linux operating system

3.Work on AMD next generation GPUs/Accelerators

4.Work on optimizing latest Rocm drivers and improve performance

5.Design new machine learning technologies

Mock Interview

Practice Video Interview with JobPe AI

Start Job-Specific Interview
cta

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

coding practice

Enhance Your Skills

Practice coding challenges to boost your skills

Start Practicing Now

RecommendedJobs for You

pune, mumbai (all areas)

hyderabad, telangana, india

nashik, maharashtra, india