AI Libraries Developer

Role overview

AI Libraries Developer at IBM

This position is for an experienced AI Library Developer with strong expertise in AI runtime frameworks and libraries development and optimization.

This role requires deep understanding of CPU architecture, Linux-based OS environments, and AI workload optimization.

Responsibilities

• You will be part of the Power System Performance team, focusing on enhancing AI frameworks and libraries to leverage the full capabilities of IBM POWER architecture.
• Ensure CPU-accelerated features are upstreamed and integrated into mainstream OPensource distributions.
• Perform performance analysis and tuning using POWER instruction profiling tools .
• Support customer-critical situations, lead proof-of-concept implementations, and provide technical guidance for pre-sales and post-sales engagements.
• You will work in a globally distributed team environment.

Employment details

• Location: Bangalore [India]

About the employer

• At IBM Infrastructure & Technology, we design and operate the systems that keep the world running.
• Our teams ensure the performance, security, and scalability that clients and industries depend on every day.
• Development experience with core libraries such as Glibc, libm, or PMDK.

Additional role details

• From high-resiliency mainframes and hybrid cloud platforms to networking, automation, and site reliability.
• Working in Infrastructure & Technology means tackling complex challenges with curiosity and collaboration.
• You’ll work with diverse technologies and colleagues worldwide to deliver resilient, future-ready solutions that power innovation.
• With continuous learning, career growth, and a supportive culture, IBM provides the opportunities to build expertise and shape the infrastructure that drives progress.
• Experience tuning for other architectures such as x86 or ARM is also highly relevant.
• As a developer, you will:
• Collaborate with internal teams and the open-source community to define, design, develop, and verify on Power-specific features in AI libraries such as OpenBLAS, ONNXRuntime, PyTorch, vLLM,Ollama and Vector Databases, with a focus on CPU kernel development and optimization.
• Develop and tune CPU kernels to accelerate AI workloads.
• Resolve issues and defects including problem recreation, debugging, proposing fixes, validating them, and working with the community to upstream solutions.
• Familiarity with software build systems, bring-up, testing, and performance analysis tools is desirable.
• Strong understanding of Computer Architecture, Operating Systems, Compilers, and Runtimes. 3+ years of experience in C/C++ and Python development. 1+ year experience in assembly language and low-level debugging. 1+ year experience in AI workload optimization.
• Experience with open-source development tools such as Git, JIRA, or Bugzilla.
• Hands-on experience with AI libraries: OpenBLAS, ONNXRuntime, PyTorch, TensorFlow, Vector DBs.
• Solid foundation in Linear Algebra, Eigenvalues, and Numerical Computing Experience in low-level programming involving OS interaction, memory management, and threading.\ Experience in GPU programming (CUDA/OpenCL) and AI workload optimization.
• Debugging, tracing, and profiling code for performance and functional issues.
• Familiarity with IBM AIX, Linux on POWER, or Z systems.
• Contributions to open-source projects or active participation in community mailing lists.
• Exposure to compiler technologies (LLVM, MLIR) and graph optimization.
• India Infrastructure & Technology Hybrid Professional Bangalore, IN Bangalore (0063) IBM India Private Limited Artificial Intelligence Assembly Language C++ Computer Architecture Debugging Git IBM Advanced Interactive Executive Atlassian Jira Linux Operating Systems Optimization Computer Programming Python PyTorch Tensorflow Compute Unified Device Architecture Open Computing Language Graphics Processing Unit Bugzilla Compiler Technologies Compilers Linear Algebra Memory Management Numerical Computing Low Level Virtual Machine