Software Engineer (Machine Learning)

Collinear AI, Inc.; Sunnyvale (CA); On-site; USD 150,000 - 250,000; Full time

Benefits offered by this job

Comprehensive medical, dental, and vision coverage

Flexible PTO

Paid parental leave

ClassPass membership

Regular in-office lunches

Job summary

Collinear AI, Inc. is seeking a Software Engineer (Machine Learning) based in Sunnyvale, California. This role focuses on developing scalable web applications and high-performance backend solutions using Python and FastAPI. Candidates should have a strong background in full stack development and experience with technologies such as React, Jenkins, and SQL databases. The position offers a competitive salary range of $150,000 to $250,000 per year, along with comprehensive benefits and equity ownership. Join Collinear to revolutionize the deployment of AI!

Qualifications

• Experience in full stack development with a focus on frontend and backend technologies.
• Strong backend development skills in Python (FastAPI) or similar languages.
• Experience with DevOps tools such as Jenkins, AWS, Docker, and Kubernetes.

Responsibilities

• Develop scalable web applications using React/Next.js.
• Design high-performance backend solutions with Python and FastAPI.
• Collaborate with teams to define features and deliver product updates.

Skills

Full stack development

Python

FastAPI

DevOps (Jenkins, AWS, Docker, Kubernetes)

SQL databases

NoSQL databases

NLP

Testing frameworks

Problem-solving

Education

Bachelor’s or Master’s degree in Computer Science/Engineering

Tools

React

Next.js

Job description

Collinear is a research-focused AI company building systems that make intelligent models more reliable, aligned, and useful in real-world settings. We work across post-training, RL, distillation, and evaluation to help teams get practical reasoning and behavior from open-source models. Our approach is grounded in technical rigor, clear thinking, and building tools that last.

How We Work

We operate with very little bureaucracy, which means decisions are made quickly and ideas move straight into action. Everyone has real ownership and is trusted to learn fast, ask questions, and figure out how to unblock themselves when the path is not obvious. We expect clear communication, thoughtful judgment, and a focus on impact rather than process. You will work closely with teammates who care deeply about the craft and the mission.

Who Thrives Here

People who enjoy solving hard problems, take initiative, and do not need hand holding tend to thrive at Collinear. If you like autonomy, fast decision cycles, and collaborating with others who hold a high bar for clarity and execution, you will feel at home here. This is a place for builders who want their work to matter and to be felt immediately.

Benefits

• Comprehensive medical, dental, and vision coverage
• 10 company holidays and flexible PTO to rest, recharge, and take care of personal needs
• 12 weeks of paid parental leave for all new parents
• ClassPass membership for fitness, wellness, and recovery
• Regular in-office lunches to make your day smoother

Learning & Growth

• Team-driven events, lunches, and opportunities to connect
• Conference support for employees with published papers
• Close collaboration with founders, research scientists, and engineering leads
• High-impact ownership across projects, features, and research areas
• Exposure to advanced LLM evaluation, post-training workflows, and emerging techniques for improving model performance
• Startup environment where your work directly shapes the company
• Competitive base salary tailored to your experience and scope of impact
• Meaningful equity ownership for full-time team members
• Relocation bonuses for candidates moving to join the team

Work with us

At Collinear, we’re revolutionizing how enterprises deploy and scale AI safely and reliably. If this resonates with you, we’d love to connect.

Software Engineer (Machine Learning)

Location

Sunnyvale, California

Employment Type

Full time

About Collinear

At Collinear, we help teams fearlessly ship AI.

Frontier labs and AI-native companies use our SimLab to find capability gaps in their agents and generate high-quality data to close them. We believe that the next generation of AI progress won’t come from just bigger models, but from more rigorous, long-horizon simulation and programmatic verification.

SimLab allows researchers to spin up realistic environments, run agents through complex tasks, and surface failure modes under real-world conditions. We then close the loop by generating targeted synthetic data to retrain models, delivering measurable quality lift on the metrics that actually matter.

About the Role

Key responsibilities include:

• Develop scalable, responsive web applications using modern frontend frameworks (React/Next.js)
• Design and implement high-performance backend solutions using Python and FastAPI, ensuring reliability and scalability
• Collaborate with cross-functional teams to define features, enhancements, and deliver product updates
• Implement and maintain DevOps best practices for continuous integration and deployment using tools like Jenkins, AWS, Docker, and Kubernetes
• Write and maintain comprehensive unit, integration, and end-to-end tests using testing frameworks
• Troubleshoot and debug frontend and backend issues, ensuring timely resolution and system optimization.
• Work with both SQL and NoSQL databases, optimizing queries for efficient data management
• Collaborate with AI/ML teams to build and deploy applications leveraging NLP and LLM technologies

About You

There are a few specific things we’ll be looking for that will help you succeed in this role:

• Bachelor’s or Master’s degree in Computer Science/Engineering, or a related field
• Experience in full stack development with a focus on both frontend and backend technologies
• Strong backend development skills in Python (FastAPI) or similar languages
• Experience with DevOps tools such as Jenkins, AWS, Docker, and Kubernetes for CI/CD pipelines
• Hands‑on experience with testing frameworks and a Test‑Driven Development (TDD) approach
• Expertise in SQL and NoSQL databases, with a solid understanding of database design and optimization
• Experience in AI/ML or NLP/LLM projects is highly desirable
• Contributions to open‑source projects, with an active GitHub portfolio showcasing innovation and expertise, are preferred
• Strong problem‑solving skills and the ability to work in a fast‑paced, collaborative environment
• Prior experience in top‑tier technology companies or startups is a plus

Collinear is an equal opportunity employer and values diversity. We do not discriminate on the basis of race, color, religion, sex, gender identity, sexual orientation, national origin, age, disability, veteran status, or any other characteristic protected by applicable law.

The base salary range for this role in California is $150,000 to $250,000 per year, depending on experience, skills, and qualifications. This role will also be eligible for equity, benefits, and bonuses.

Collinear provides reasonable accommodations for candidates with disabilities throughout the application and hiring process. If you need an accommodation, please contact us.

Pursuant to applicable local ordinances, we will consider qualified applicants with arrest and conviction records.