Role Summary

Role:ML Research Engineer (Intern / Full-Time)

Company: Sweep

Location: San Francisco, California (On-site)

Job Type: Internship / Full-Time

Compensation: $180,000 – $300,000 USD/year (Full-Time) + 1.00% – 4.97% Equity | Competitive Internship Stipend

Visa Sponsorship: Available for eligible candidates

About Sweep

Sweep is building one of the world's most advanced AI coding assistants for JetBrains IDEs, helping developers write, understand, and ship code faster. The company develops its own coding language models, builds custom LLM fine-tuning and inference infrastructure, and creates intelligent developer tools that improve software engineering productivity at scale.

If you're passionate about machine learning, large language models, and developer tools, this is an opportunity to work on cutting-edge AI systems alongside a world-class engineering team.

Key Responsibilities

• Research, fine-tune, and evaluate Large Language Models (LLMs) for code generation and software engineering tasks.
• Build and optimize AI models for code completion, editing, and autonomous coding workflows.
• Develop scalable backend services and inference pipelines using Python.
• Improve model quality, latency, reliability, and developer experience through experimentation and evaluation.
• Build AI-powered features for JetBrains IDEs using Kotlin.
• Design experiments, benchmark model performance, and iterate on new research ideas.
• Collaborate closely with engineers to ship production-ready AI features.
• Contribute to the development of next-generation AI coding assistants used by developers worldwide.

Eligibility

• Pursuing or recently completed a Bachelor's, Master's, or PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• Strong programming skills in Python.
• Solid understanding of machine learning, deep learning, and data structures.
• Strong analytical thinking and problem-solving abilities.
• Ability to work independently and thrive in a fast-paced startup environment.

Preferred Skills

• Experience working with Large Language Models (LLMs), transformers, or model fine-tuning.
• Familiarity with machine learning frameworks such as PyTorch or TensorFlow.
• Knowledge of Kotlin or JVM-based development.
• Experience building developer tools, IDE plugins, or coding assistants.
• Understanding of distributed systems, model serving, or inference optimization.
• Contributions to open-source projects, AI research, coding competitions, or personal ML projects.

Tech Stack

• Languages: Python, Kotlin
• Focus Areas: LLM Fine-tuning, Model Inference, AI Coding Assistants
• Products: JetBrains Plugin Development, AI Developer Tools

What You'll Gain

• Work on one of the most advanced AI coding assistants in the industry.
• Gain hands-on experience building production-grade coding LLMs and AI infrastructure.
• Collaborate directly with experienced AI researchers and engineers.
• Own impactful projects from research to deployment.
• Learn how state-of-the-art AI products are built and scaled.
• Opportunity to transition into a full-time engineering role based on performance.

Interview Process

The interview process includes a practical coding challenge where candidates build a GitHub Copilot-style autocomplete plugin for JetBrains IDEs using the IntelliJ Platform Plugin Template. Shortlisted candidates will participate in technical discussions focused on machine learning, software engineering, system design, and problem-solving.

If you're excited about building the future of AI-powered software development and working at the intersection of machine learning and developer productivity, we'd love to hear from you.