AI ML Engineer (Freshers)

Location: Mandaveli, Chennai
Experience: 0-1 Years
Salary: 5-7 Lacs P.A.
Vacancies: 15
Batch: 2025, 2026
Employment Type: Full Time

About the Role

Design and develop compiler frameworks that optimize AI model execution. Build end-to-end AI pipelines including graph optimizations, memory scheduling, and compute distribution. Work on transformer-based architectures for distributed multi-node training and efficient inference.

Education

Bachelor's or Master's in Computer Science, AI, Data Science, ML, Mathematics, or Statistics

Key Responsibilities

  • Design and develop compiler frameworks that optimize AI model execution at the kernel, graph, and operator levels
  • Architect scalable transformer-based infrastructures for distributed multi-node training and efficient inference
  • Build end-to-end AI pipelines including graph optimizations, memory scheduling, and compute distribution
  • Collaborate with research teams to translate mathematical models into optimized execution graphs and intermediate representations (IRs)
  • Implement custom kernels, quantization strategies, and low-level performance optimizations in C/C++ and CUDA
  • Analyze and tune runtime performance bottlenecks focusing on parallelization, vectorization, and memory management
  • Develop domain-specific compiler passes for tensor operations, automatic differentiation, and operator fusion
  • Conduct systematic experiments to explore scaling laws, precision formats, and architectural optimizations

Requirements

  • Strong proficiency in C, C++ or Java, with good command over pointers, memory management, performance optimization, and systems-level programming
  • Solid foundation in Mathematics - calculus, probability, statistics, and linear algebra
  • Strong logical reasoning, problem-solving ability, and an analytical mindset
  • Ability to operate effectively in a research-driven, high-intensity environment with cross-functional collaboration

Added Advantage

  • Experience or strong interest in compiler construction, runtime systems, and code generation
  • Proficiency or willingness to learn CUDA and Rust for high-performance and systems-level development
  • Deep understanding of computer architecture, operating systems, data structures, and memory management

Why Join BluBridge?

  • A dynamic and innovative work environment with an AI research-driven team
  • Mentorship, learning, and growth opportunities in AI infrastructure and systems research
  • Competitive compensation with performance-based advancement and technical ownership
  • Opportunity to work on compiler-level AI framework design and large-scale foundation model optimization
  • Access to multi-node GPU clusters and next-gen compute environments for experimentation

Ready to Join Our Team?

Apply now and be part of BluBridge's mission to build the future of AI.