Overview
About the job:
Key responsibilities:1. Design and develop compiler frameworks that optimize AI model execution at the kernel, graph, and operator levels.
2. Architect scalable transformer-based infrastructures for distributed multi-node training and efficient inference.
3. Build end-to-end AI pipelines including graph optimizations, memory scheduling, and compute distribution.
4. Collaborate with research teams to translate mathematical models into optimized execution graphs and intermediate representations (IRs).
5. Implement custom kernels, quantization strategies, and low-level performance optimizations in C/C++ and CUDA.
6. Analyze and tune runtime performance bottlenecks focusing on parallelization, vectorization, and memory management.
7. Develop domain-specific compiler passes for tensor operations, automatic differentiation, and operator fusion.
8. Conduct systematic experiments to explore scaling laws, precision formats, and architectural optimizations for improved computational efficiency.
Who can apply:
- are Computer Science Engineering students
Only those candidates can apply who:
Salary:
₹ 5,00,000 - 7,00,000 /yearExperience:
0 year(s)Deadline:
2026-02-21 23:59:59Skills required:
Java, Python, Machine Learning, Data Science, C++ Programming, Neural Networks, Natural Language Processing (NLP), Deep Learning, TensorFlow, Artificial intelligence and PyTorchOther Requirements:
Mandatory requirements:
1. Strong proficiency in C, C++ or Java, with good command over pointers, memory management, performance optimization, and systems-level programming.
2. Solid foundation in Mathematics - calculus, probability, statistics, and linear algebra.
3. Strong logical reasoning, problem-solving ability, high intelligence (IQ), and an analytical mindset.
4. Ability to operate effectively in a research-driven, high-intensity environment with cross-functional collaboration.
Added advantage:
1. Experience or strong interest in compiler construction, runtime systems, and code generation.
2. Proficiency or willingness to learn CUDA and Rust for high-performance and systems-level development.
3. Deep understanding of computer architecture, operating systems, data structures, and memory management.
Important: Please read the guidelines clearly before applying: https://blubridge.ai/join-our-team
Role Category: Data science & machine learning
Education:
1. UG: B.Tech/B.E. in Computers
2. PG: M.Tech in Any Specialization, MS/M.Sc(Science) in Maths
3. Doctorate: Any Doctorate
About Company:
Our mission at Blubridge is to develop deep expertise in modeling artificial intelligence systems, their applications, and research. We are exploring optimizing tokenization methods, enhancing attention mechanisms, and refining AI systems for chatbots, audio, and image processing.