Research Engineer Job in International Business Machines

Research Engineer

Location: Bangalore, Karnataka, India

Job Type: Full-Time

Experience Level: 0-8 years

Company: IBM Research India (IRL)

Introduction:

IBM Research is the innovation engine of IBM and is the largest industrial research organization in the world. With 12 labs across 6 continents and over 3200 researchers globally, we produce more patents daily than any other organization. At IBM Research India (IRL), we are shaping the future of computing in areas like AI, Hybrid Cloud, and Quantum Computing. Our work is at the forefront of breakthrough innovations in Foundation Models, AI systems, large-scale data engineering, and more.

We are looking for top talent to join us in our exciting and dynamic projects, pushing the boundaries of innovation. As a Research Engineer, you will work on pioneering research and development in the most cutting-edge fields of AI and computing.

Role Overview:

The Research Engineer role at IBM India Research Lab (IRL) involves working on challenging, dynamic, and highly innovative projects in the fields of AI, machine learning, and data systems. Your responsibilities will span multiple areas including optimizing AI models for large-scale distributed systems, pre-training foundation models, and developing real-world use cases that leverage IBM s infrastructure and models.

Key Responsibilities:

Optimized Runtime Stacks for Foundation Models:

• Work on fine-tuning, inference serving, and large-scale data engineering for AI models.
• Focus on multi-stage tuning, reinforcement learning, inference-time compute, and preparing data for complex AI systems.

Model Optimization Across Accelerators:

• Develop solutions to optimize models for multi-accelerator environments, particularly focusing on IBM s AIU accelerator.
• Work on compiler optimizations, specialized kernels, libraries, and tools to enhance model performance.

Pre-training and Deployment of Foundation Models:

• Participate in pre-training language models and multi-modal foundation models.
• Work on distributed training procedures, model alignment, and creating pipelines for various tasks, including LLM-generated data pipelines.

Research and Use Case Development:

• Develop and implement use cases that effectively leverage infrastructure and models to drive real-world value.
• Contribute to creating frameworks for human-data collection and deploying models on user-centric platforms.

Required Education and Experience:

Education:

• A Master s degree in Computer Science, AI, or related fields from a top institution.

Experience:

• 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, and inference optimizations.

Technical Skills:

• Experience with big data platforms such as Ray and Spark.
• Experience with Pytorch FSDP and HuggingFace libraries.
• Proficiency in programming with Python or web development technologies.

Mindset and Attitude:

• A growth mindset and pragmatic approach to problem-solving.

Preferred Experience:

Research Experience:

• Peer-reviewed research at top machine learning or systems conferences.

Advanced Technical Skills:

• Experience working with pytorch.compile, CUDA, Triton kernels, GPU scheduling, and memory management.

Open Source Contributions:

• Experience working within open-source communities, contributing to or developing open-source projects.

Why IBM Research?

• Innovative Environment: Be at the forefront of technological innovation, working on cutting-edge projects in AI, quantum computing, and more.
• Global Impact: Work on projects that influence both academic research and commercial product development, making a global impact.
• Career Development: IBM offers abundant opportunities for learning and growth, with access to the latest technologies and research.
• Collaborative Culture: Work with a diverse team of world-class researchers and engineers in a collaborative, open-source-driven environment.

Ready to join IBM Research and help shape the future of AI and computing?

Apply today and become a part of the team that s redefining innovation.