Intern Researcher - Agentic ML Compiler

yasp is pioneering the future of software development with a compiler that leverages agentic AI for advanced optimization and code generation. We are looking for an **Intern Researcher - Agentic ML Compiler** to join our team and help drive the core innovation that will define the next generation of our technology. We don’t draw boundaries between research and engineering. We are creative thinkers and relentless prototypers who live at the bleeding edge of AI research and engineering. Together, we explore, invent, and ship novel technology that pushes beyond the state of the art for accelerating deep learning models. Tasks Develop novel agentic AI and LLM-based methods to optimize kernels for machine learning models. Evaluate the performance of AI driven ML compilation techniques via rigorous benchmarking. Stay up to date with the publication landscape, open-source projects, and industry trends. Contribute to writing research papers for publication in top AI conferences. Work with a talented team towards breakthroughs in AI for ML compilation. Requirements Master’s or PhD student in Computer Science/Maths/Physics with a focus on AI/ML, or equivalent practical experience. Exceptional prior work in AI/ML, ideally involving LLMs or agents to solve complex tasks. Strong engineering and coding abilities (PyTorch, agentic AI frameworks, LLM post-training frameworks, software development best practices). Passion for good scientific practice, blending clean formal thinking and good communication with exacting engineering. Preferred: Experience post-training LLMs using SFT and RL, especially as part of an AI agent. Knowledge of ML compiler principles including intermediate representations, operator fusion, and custom CUDA or Triton kernel generation. Previous publications in leading AI conferences (e.g., NeurIPS, ICLR, ICML). Benefits Competitive salary. Comprehensive health benefits (including dental). Opportunities for professional development and growth. Flexible work hours. Dynamic and collaborative work environment. Cutting-edge software and hardware platforms. Ready to apply? Submit your application through this posting and our team will review it personally. From there, you can expect a short intro call, a technical evaluation built around real problems, and conversations with the people you'd work with directly.

Develop novel agentic AI and LLM-based methods to optimize kernels for machine learning models. Evaluate performance through rigorous benchmarking and contribute to research papers for top AI conferences.

Requires a Master's or PhD student in Computer Science, Maths, or Physics with a focus on AI/ML. Candidates should have strong coding abilities in PyTorch and experience with LLMs or AI agents.

• Competitive Salary • Comprehensive Health Benefits • Dental Benefits • Professional Development And Growth • Flexible Work Hours • Dynamic And Collaborative Work Environment • Cutting-edge Software And Hardware Platforms

Relevées dans la description du poste. Confirmez les exigences importantes ci-dessus.

• PyTorch
• Agentic AI Frameworks
• LLM Post-training Frameworks
• CUDA
• Triton
• ML Compilation
• SFT
• RL
• Kernel Optimization
• Benchmarking
• Software Development

Formation minimale

Autre formation

Expérience minimale

0+ ans