Source Job

5w PTO 26w maternity 2w paternity

  • Lead architecture and delivery of distributed cloud-native applications for high concurrency and real-time data needs.
  • Design production-grade data pipelines and integrate ML/GenAI capabilities, including RAG and model serving.
  • Mentor peers, set engineering standards, and evolve legacy systems into microservices for scale.

Go Python Kafka Data Engineering AI/ML

20 jobs similar to Staff Software Engineer - AI

Jobs ranked by similarity.

  • Own the architecture, design, and delivery for your team's platform area, turning roadmap into engineering plans that ship and hold up in production.
  • Design and build core parts of a compound AI system spanning agents, LLMs, and RAG, with a focus on reliability, evaluation, and scale.
  • Champion engineering best practices, mentor engineers, and partner with product and design counterparts to align on priorities.

Crogl builds fully autonomous AI team members for security operations centers (SOCs), helping organizations secure their digital estates. As a startup with a focus on innovation, they seek senior technical leaders to join their remote team.

Global

  • Lead technical direction for AI-driven application-builder architecture, design systems, and mentor engineers.
  • Architect and build scalable, fault-tolerant LLM pipelines and distributed systems handling high-throughput generative workloads.
  • Drive quality, observability, and reliability with SLOs, monitoring, and hallucination mitigation strategies.

HighLevel is an AI-powered business operating system that gives agencies, entrepreneurs and SMBs the infrastructure to build, automate and scale. With over 2,000 team members across 10+ countries, we operate as a global, remote-first organization built for speed and ownership.

India

  • Lead the technical architecture and evolution of core AI-powered application-building services, ensuring scalability and reliability.
  • Design and build distributed backend systems using Go, Node.js, PostgreSQL, and cloud technologies like Cloudflare Workers.
  • Implement AI safety mechanisms, mentor engineers, and collaborate cross-functionally to drive technical excellence.

Jobgether is an AI-powered job matching platform that connects candidates with hiring companies. They use automated systems to review applications and share shortlists, fostering an inclusive and efficient recruitment process.

United States Unlimited PTO

  • Design and build scalable backend systems powering AI agents in real-time enterprise environments.
  • Develop agent orchestration frameworks and low-latency inference pipelines integrating LLMs and SLMs.
  • Build robust APIs and work with cross-functional teams to productionize agentic AI at scale.

Level AI is an AI-native platform that helps enterprises transform contact centers into engines of customer intelligence and operational efficiency. The company is a Series C startup backed by Battery Ventures and ENIAC, based in Mountain View, California, with a globally distributed team.

$221,200–$387,100/yr
North America

  • Architect and build AI-native products for enterprise service operations.
  • Lead technical direction and design scalable distributed systems.
  • Mentor engineers and drive engineering excellence across teams.

ServiceNow is the AI control tower for business reinvention, helping 85% of the Fortune 500 work smarter, faster, and better. The company fosters an AI-native culture where technology and talent are unstoppable together.

EMEA

  • Design and build large-scale distributed systems and high-throughput data pipelines using Go and cloud-native technologies.
  • Lead system-wide architectural decisions focusing on data flow, performance, and resilience.
  • Champion best engineering practices, code quality, testing, and maintainability while mentoring junior engineers.

DoiT is a global technology company that helps organizations leverage the cloud for business growth, combining data, technology, and human expertise. With thousands of customers worldwide, DoiT fosters a remote-first culture that values entrepreneurship, knowledge pursuit, and fun.

$210,000–$260,000/yr
US Canada India Luxembourg England Unlimited PTO

  • Own end-to-end delivery of core data platform components including schema mapping, normalization, and validation pipelines.
  • Drive technical architecture for an AI-native data warehouse serving institutional financial clients.
  • Build AI evaluation infrastructure to ensure trustworthy outputs in high-stakes financial data contexts.

Juniper Square is a private market operations platform that unifies technology, data, and fund administration services for over 2,300 GPs. With 1,000+ employees and $350M+ in funding, the company has a founder-led culture focused on ambitious, meaningful work and diverse perspectives.

US

  • Lead technical architecture and engineering standards across customer-facing and operational systems.
  • Champion AI-assisted development, establishing best practices for LLMs in coding, testing, and workflows.
  • Mentor engineers and drive engineering excellence through code quality, CI/CD, and operational reliability.

Koala Health aims to make managing your pet’s health simple and delightful, offering a pharmacy service for medications and supplements delivered in pre-sorted pouches. The company fosters a remote-first culture built on flexibility and trust, with a supportive, close-knit team that celebrates wins and learns fast.

Europe

  • Take full ownership of INFOnline's central data platform from raw event ingress through reporting delivery, defining the roadmap and driving execution.
  • Establish engineering standards, mentor others, and drive AI-native engineering practices to increase speed and quality.
  • Collaborate with Product, Customer Success, and Leadership to translate business requirements into scalable technical solutions.

INFOnline powers digital audience measurement for the German and Austrian media industry, processing billions of events to deliver trusted reach and engagement metrics. Part of saas.group, they are a small, friendly team focused on modernizing infrastructure and moving to a cloud-native architecture on GCP.

Global

  • Architect and scale data systems for analytics, ML/AI products, reporting, and APIs.
  • Own the full data lifecycle including ingestion, transformation, modeling, validation, and serving.
  • Partner with Data Science to productionize models and build reliable data foundations for AI-driven products.

Vidmob is the creative data company that provides scoring software and analytics to help marketers and agencies drive business results through improved creative effectiveness. They partner with top marketers and agencies worldwide and operate the industry's most robustly instrumented human-reinforcement learning model for creativity.

Global

  • Own data architecture at Buildkite, setting the strategy and technical direction for the entire data effort.
  • Design for agents and humans both, making data discoverable, trustworthy, and safe to act on.
  • Choose the stack—query engines, semantic layer, ingestion, streaming, and storage—rather than inherit it, and stay hands-on where it counts.

Buildkite provides a continuous integration and deployment platform for software teams, with a focus on developer tools and CI/CD. The company is a remote-first organization of about 150 people, fostering a flat, high-autonomy culture that values deep focus and genuine team overlap.

US

  • Serve as primary technical advisor to customers, leading discovery sessions and translating business priorities into AI-enabled solutions.
  • Lead cross-functional engineering teams to design and deploy secure, scalable, and high-performing AI software systems.
  • Mentor engineers, establish engineering standards, and drive delivery of enterprise-scale AI solutions.

LTS delivers AI-powered software solutions for complex business and mission challenges. The company values innovation, growth, collaboration, and quality, and offers a culture that rewards ambition and performance.

US

  • Write behavioral specs, architectural constraints, and feature requirements that agents implement against.
  • Build and maintain harness infrastructure including structural tests, linting rules, and CI gates.
  • Design validation systems where agents write the tests and you verify features work from the user's perspective.

Bolo.ai builds generative AI systems for the energy industry, making daily work faster, safer, and better for heavy industry workers. We have Fortune 500 contracts, production deployments, and growing enterprise demand, and we're scaling with a small, senior-leaning engineering team.

US 16w maternity 12w paternity

  • Orchestrate High-Velocity Workflows: Leverage advanced agentic coding tools (e.g., Cursor, multi-agent environments) to dramatically accelerate feature prototyping, code generation, and test coverage.
  • Own the Guardrails & Quality: Act as the ultimate reviewer and architect; define the specifications, establish repo-context guardrails, and review AI-accelerated output for hidden security risks, scale bottlenecks, and architectural alignment.
  • Build Scalable Application and Data Layers: Design, build, and maintain our data pipelines and application to service our hundreds of users.

EvolutionIQ provides technology to improve insurance claims handling. The company is experiencing massive growth and has been named a top workplace, prioritizing its team.

$195,000–$220,000/yr
US Canada

  • Assess current pipelines and data architecture to produce a prioritized plan for change.
  • Design durable data and ML systems grounded in customer needs with documented tradeoffs.
  • Harden pipelines, upgrade data architecture, and raise standards for observability and reliability.

FutureFit AI's core mission is to help more people get to better jobs faster and cheaper, with a focus on those facing barriers to opportunity. Their team of 30-50 across the US and Canada fosters a high trust, high intensity culture with a will to win.

US Unlimited PTO 16w maternity 16w paternity

  • Design, build, and deploy AI-powered solutions connecting Remote’s platform to customer systems and workflows.
  • Work at the frontier of practical AI, shipping reliable, observable, and secure systems in production.
  • Own customer outcomes from discovery through production rollout with high autonomy and direct influence on product roadmap.

Remote is a global HR platform that helps businesses recruit, pay, and manage international teams compliantly. The company fosters a future-focused, async culture with team members across 6 continents and emphasizes innovation, automation, and AI.

India

  • Lead a team of senior engineers building AI-first GTM systems using Python, TypeScript, and AWS.
  • Own the engineering roadmap and drive delivery of agentic AI pipelines and Salesforce integrations.
  • Champion engineering quality and a culture of responsible AI development with human-in-the-loop controls.

Samsara is the pioneer of the Connected Operations Cloud, enabling organizations to harness IoT data to improve safety, efficiency, and sustainability. As a recently public company, we employ thousands and foster a culture of high-impact builders with an employee-led remote model.

$120,000–$185,000/yr
US

  • Lead the development of new mechanisms for monitoring the health of data from edge to cloud.
  • Drive architectural decisions and technical strategy across products and platforms.
  • Act as a technical mentor for engineers at all levels and contribute to key codebases.

Johnson Controls is a global leader in thermal management, mission-critical building systems, energy efficiency, and decarbonization. With over 140 years of experience, the company employs a large workforce and fosters a culture of innovation and sustainability.

$140,000–$165,000/yr
US

  • Design, build, and maintain scalable back-end systems and APIs for high-throughput messaging operations.
  • Improve platform reliability, performance, observability, and automation across onboarding, compliance, and routing.
  • Collaborate with Product and Operations teams to translate complex workflows into scalable distributed systems.

Volt Labs builds the infrastructure layer behind modern business messaging, helping companies navigate messaging compliance, onboarding, deliverability, routing, and operational complexity across carriers and providers. We are a remote-first, high-ownership team operating in a fast-moving growth environment with a deep focus on builder mentality, operational excellence, and collaborative culture.

Global

  • Design, build, and evolve Data Warehouses, Data Marts, and enterprise data models to support analytics and business decision-making.
  • Design, implement, and optimize scalable ELT pipelines and data processing solutions, ensuring reliability, performance, and maintainability.
  • Drive technical decisions related to data architecture, platform evolution, data modeling approaches, and integration patterns.

Sigma Software is building a scalable, reliable, and secure corporate Data Platform to power analytics and decision-making across the company. The company supports professional growth, knowledge sharing, and modern tools for high-quality solutions.