Our client, a venture-backed AI company, is hiring an AI Engineer to join their R&D team in New York. The successful candidate will contribute to the development of cutting-edge AI technologies, working on innovative projects that integrate the latest advancements in machine learning and code generation.
Responsibilities
- Investigate and evaluate the performance of different LLMs in tool usage and code generation.
- Create tools and APIs to improve LLM capabilities in generating and enhancing GPU kernels.
- Develop programming solutions using C/C++ and Python.
Skillset
- Proficient in programming with C/C++ and Python.
- Experience in developing AI agents that utilize LLMs and external tools.
- Familiarity with training and deploying ML models.
- Bonus points for demonstrated experience with kernel optimization on CUDA, ROCm, or other accelerators.
- In-depth knowledge of GPU performance optimization techniques is a plus.
Benefits
- Salary: Up to $200k depending on experience.
- Equity.
- Health insurance for you and your family.
Interested? Apply Today!