Skip to content
RollstackRollstack

AI Software Engineer

Build production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.

About the job

Responsibilities

  • Build AI-powered features that automate data sharing through slides and documents.
  • Use established frameworks and innovative techniques to create scalable, production-ready solutions for data analysis and content generation.
  • Improve software reliability and performance to accelerate delivery and meet customer needs.
  • Collaborate with designers and software engineers to increase product adoption and engagement.
  • Develop best practices with emerging AI tools and techniques.

Technical Environment

  • TypeScript, React, Tailwind CSS, and Shadcn/UI
  • Node.js, Prisma ORM, and Temporal workflows
  • Kubernetes on AWS with Argo CD
  • SigNoz, Sentry, and PostHog
  • OpenAI API, Gemini, LangChain, and Langfuse
  • Linear

Requirements

  • 3+ years of experience in machine learning and natural language processing, with a focus on generative models.
  • Advanced Python programming skills.
  • Proficiency with large language model frameworks, including LangChain, Retrieval-Augmented Generation (RAG), and LlamaIndex.
  • Proven experience researching, building, and fine-tuning large language models in production environments.

Nice to Have

  • Experience building LLM agents for data analysis and content generation.

Compensation and Benefits

  • Competitive compensation and meaningful equity.
  • Fully remote, globally diverse workplace.
  • Annual company gathering in a new city.

Skills

Python, LangChain, Retrieval-Augmented Generation, Llamaindex, LLMs, Natural Language Processing, Machine Learning, Llm Agents, TypeScript, React, Node.js, Prisma Orm, Temporal, Kubernetes, AWS

Thinking Machines Lab

Thinking Machines Lab

San Francisco, CA

Research Software Engineer, Post Training
$350k+/yrHybridML Engineering

Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.

OpenAI

OpenAI

San Francisco, CA

Software Engineer, AI for Chip Design
$266k+/yrHybridML Engineering

Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.

ClickUp

ClickUp

United States

Machine Learning Engineer, Ranking & Retrieval
$200k+/yrRemote5+ YOEML Engineering

Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.

PathAI

PathAI

Boston, MA
Machine Learning Engineer III
$131k+/yrOn-site5+ YOEML Engineering

Develop and deploy machine learning models for biomedical research and AI products, collaborating with scientific, engineering, and product teams. Requires an advanced quantitative degree, substantial ML experience, Python proficiency, and experience bringing models into production or research applications.

Mirage

Mirage

New York, NY

Software Engineer, Agents
$175k+/yrOn-site5+ YOEML Engineering

Build and deploy agentic systems that power AI-driven creative video workflows. The role requires 5+ years of experience, production ML or agentic pipeline development, context engineering, and expertise in evaluation and agent infrastructure.