remote
AI Engineer - Makora
AI Engineer
AI Engineer to build a next‑generation machine learning compiler that fuses code‑generating LLMs into the compilation pipeline, designing and evaluating AI agents for automated algorithm discovery.
About the role
Key Responsibilities
- Design, implement, and optimize a machine learning compiler that integrates code‑generating large language models into the compilation flow.
- Develop and evaluate AI agents that automate algorithm discovery and code generation for performance‑critical workloads.
- Collaborate with research and engineering teams to prototype new compiler passes and LLM‑based optimizations.
- Benchmark and analyze compiler performance, ensuring scalability and correctness across diverse hardware targets.
- Document architecture, design decisions, and experimental results for internal and external dissemination.
Requirements
- Strong background in Python and machine learning frameworks (PyTorch, TensorFlow).
- Experience with large language models, especially code‑generation models such as GPT‑4, CodeGen, or similar.
- Solid understanding of compiler theory, intermediate representations, and code optimization techniques.
- Proficiency in building and evaluating AI agents for automated reasoning and decision making.
- Excellent problem‑solving skills and ability to work independently in a fast‑paced research environment.
Skills
pythonmachine learning