Augment Code is the only AI coding assistant built for professional software engineers working in large, production‑grade codebases. Our Context Engine understands your entire repo, enabling developers to stay in flow while writing, reviewing, and understanding code. Backed by top‑tier investors and trusted by engineering teams at leading tech companies, Augment Code is redefining how modern software is built.
About the RoleAs a Research Engineer at Augment Code, you’ll sit at the intersection of research and engineering - transforming cutting-edge ideas into production-ready systems that redefine how developers build software. You’ll design models that understand entire codebases, create the infrastructure to support large-scale experiments, and ship innovations that directly power our Context Engine and developer productivity platform.
In this role you will:- 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.
- Degree in Computer Science, Machine Learning, or related technical field (or equivalent industry experience).
- 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.
- Flexible work hours
- Competitive salary & Equity
- Tools Stipend
- Health, Dental, Vision and Life Insurance
- Short Term and Long Term Disability
- Unlimited Paid Time Off + Holidays. We focus on trust and ownership, not time in the chair
- Numerous company social events
We will do everything we can within reason to make sure that your interview takes place in an environment that fairly and accurately assesses your skills. If you need assistance or accommodation, please contact your recruiter.
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.
By applying for this job, the candidate acknowledges and agrees that any personal data contained in their application or supporting materials will be processed in accordance with Augment Code's Applicant Privacy Policy.
Pay Transparency Notice: The actual base salary within the stated range will be based on a combination of factors such as an individual's skills, experience level, educational background, and other relevant job-related considerations.
Top Skills
What We Do
Augment puts your team’s collective knowledge—codebase, documentation, and dependencies—at your fingertips via chat, code completions, and suggested edits.
Get up to speed, stay in the flow and get more done. Lightning fast and highly secure, Augment works in your favorite IDEs and Slack.
We proudly augment developers at Webflow, Kong, Pigment, and more.
We are alumni of great AI and cloud companies, including Google, Meta, NVIDIA, Snowflake, and Databricks.
If, like us, you believe in augmenting and not replacing software developers, join us on our mission to improve software development at scale using AI.
