Highlights
Dynamic environment, mentorship, growth opportunities, cutting-edge research, technical ownership.
Description
Join our cutting-edge research and development team as an AI & ML Engineer/Researcher. This role is ideal for fresh graduates looking to make a significant impact in the field of artificial intelligence. As part of our vibrant and dynamic environment, you will collaborate with experts in data science, machine learning, and computer science to design and develop advanced AI models.
Key Responsibilities:
- Design compiler frameworks that optimize AI model execution at the kernel, graph, and operator levels.
- Arcitect scalable transformer-based infrastructures for distributed multi-node training and efficient inference.
- Create 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 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 for improved computational efficiency.
Mandatory 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.
- High logical reasoning, problem-solving ability, and an analytical mindset.
- Aptitude for working in a high-intensity, research-driven environment with cross-functional collaboration.
We offer:
- A dynamic work environment with mentorship and growth opportunities in AI infrastructure and systems research.
- Competitive compensation with performance-based advancement and technical ownership.
- The chance to contribute to the design of compiler-level AI frameworks and optimize large-scale foundation models.
Skills
