Software Engineer (Machine Learning)

Collinear AI; United States; On-site; USD 150,000 - 250,000; Full time

Benefits offered by this job

Equity

Benefits

Bonuses

Job summary

Collinear AI is seeking a Software Engineer (Machine Learning) to build scalable web applications using React/Next.js on the frontend and Python/FastAPI on the backend. You will work with DevOps tools to enable CI/CD pipelines and deploy AI/ML/NLP-enabled features.

The ideal candidate has hands-on experience with React/Next.js, Python, FastAPI, and containerized deployments in Jenkins/AWS/Docker/Kubernetes environments, plus a passion for testing and collaboration.

Qualifications

• Full stack development with a frontend/backend focus.
• Strong experience with React/Next.js and Python (FastAPI).
• DevOps exposure with CI/CD pipelines and containerization.
• Experience with testing frameworks and TDD.
• Experience collaborating with cross-functional teams and AI/ML/NLP projects.

Responsibilities

• Develop scalable, responsive web applications using React/Next.js.
• Build backend services with Python and FastAPI, ensuring reliability and scalability.
• Collaborate with cross-functional teams to define features and updates.
• Implement CI/CD with Jenkins, AWS, Docker, and Kubernetes.

Skills

Frontend development

Backend development

TDD testing

Problem solving

Collaboration

Open source contributions

Education

Bachelor’s or Master’s in CS/Engineering

Tools

React

Next.js

FastAPI

Jenkins

AWS

Docker

Kubernetes

SQL

NoSQL

GitHub

Job description

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

We are looking for a talented Software Engineer (Machine Learning) with expertise in React and NextJs (JavaScript frameworks) for frontend development and backend development in Python and Fast API. The ideal candidate will have hands‑on experience in DevOps technologies, testing frameworks, database management, and exposure to AI/ML or NLP/LLM projects.

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

• Proficiency in JavaScript frameworks (React/Next.js) for frontend development

• 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.