Augment Code is looking to solve the problem of how to build AI coding assistants that can understand large, production-grade codebases, enabling developers to stay in flow while writing, reviewing, and understanding code. The Research Engineer will transform cutting-edge ideas into production-ready systems that redefine how developers build software.
Requirements
- Deep experience with LLMs, Retrieval Systems, or Post-Training.
- Strong understanding of modern machine learning techniques (reinforcement learning, transformers, etc).
- Experience with large-scale data processing, training, and model deployment.
- Proficiency with deep learning frameworks (e.g., PyTorch, TensorFlow).
- Contributions to open-source projects or publications in relevant fields.
- Background in developer tools, IDEs, or software engineering productivity systems.
Responsibilities
- Design and implement ML models and algorithms that advance code understanding, code classification, retrieval, and developer productivity.
- Own projects end-to-end: from ideation and prototyping to optimization and production deployment.
- Build and maintain research infrastructure, experimentation frameworks, and data pipelines at scale.
- Collaborate closely with engineering and product teams to translate research into user-facing impact.
- Stay up-to-date with the latest developments in machine learning, natural language processing, and software engineering research.
Other
- Degree in Computer Science, Machine Learning, or related technical field (or equivalent industry experience).
- Flexible work hours
- Unlimited Paid Time Off + Holidays. We focus on trust and ownership, not time in the chair
- Augment Code is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon race, religion, color, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics.
