A cutting-edge AI infrastructure company is seeking an AI engineer with experience in code-generating LLMs and AI agent development. They are developing a next-gen machine learning compiler that integrates cutting-edge LLMs into the compilation process
Responsibilities:
- Evaluate LLMs for code generation and tool use
- Build tools/APIs to enhance GPU kernel generation
- Develop in C/C++ and Python
Requirements:
- Strong coding skills (C/C++, Python)
- Experience with LLM-powered AI agents
- Knowledge of LLM training, fine-tuning, and deployment
Bonus:
- GPU/kernel optimization experience (CUDA, ROCm, etc.)
- Deep GPU performance expertise
