New York

  Machine Learning

Permanent

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

  • As the AI Engineer, you will 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!

49937