Job Description
Research Engineer position at IBM India Research Lab is a challenging, dynamic and highly innovative role. Some of our current areas of work where we are actively looking for top talent are: Optimized runtime stacks for foundation model workloads including fine-tuning, inference serving and large-scale data engineering, with a focus on multi-stage tuning including reinforcement learning, inference-time compute, and data preparation needs for complex AI systems. Optimizing models to run on multiple accelerators including IBMs AIU accelerator leveraging compiler optimizations, specialized kernels, libraries and tools. Developing use cases that effectively leverage the infrastructure and models to deliver value Pre-training language and multi-modal foundation models working with large scale distributed training procedures, model alignment, creating specialized pipelines for various tasks including effective LLM-generated data pipelines, creating frameworks for collecting human data and deploying models in user-centric platforms. Required education
Bachelor's Degree Preferred education
Master's Degree Required technical and professional expertise
You should have one or more of the following: A masters degree in computer science, AI or related fields from a top institution 0-8 years of experience working with modern ML techniques including but not limited to model architectures, data processing, fine-tuning techniques, reinforcement learning, distributed training, inference optimizations Experience with big data platforms like Ray and Spark Experience working with Pytorch FSDP and HuggingFace libraries Programming experience in one of the followingPython, web development technologies Growth mindset and a pragmatic attitude Preferred technical and professional experience
Peer-reviewed research at top machine learning or systems conferences Experience working with pytorch.compile, CUDA, triton kernels, GPU scheduling, memory management Experience working with open-source communities
Job Classification
Industry: IT Services & Consulting
Functional Area / Department: Research & Development
Role Category: Pharmaceutical & Biotechnology
Role: Research Associate / Engineer
Employement Type: Full time
Contact Details:
Company: IBM
Location(s): Bengaluru
Keyskills:
cuda
memory management
spark
pytorch
ml
image processing
python
c++
c
natural language processing
data processing
neural networks
machine learning
lab
artificial intelligence
deep learning
tensorflow
java
computer vision
opencv
pattern recognition