Source Job

Global 6w PTO 26w maternity 26w paternity

  • Design and write high-performing, scalable software for training models.
  • Develop new tools to support and accelerate research and LLM training.
  • Collaborate with engineering and scientific teams to create a strong post-training ecosystem.

Python JAX PyTorch Distributed Training Software Engineering

20 jobs similar to Member of Technical Staff, Integration/RL Team (Research Engineer)

Jobs ranked by similarity.

Global 6w PTO 26w maternity 26w paternity

  • Design and write high-performing scalable software for training models.
  • Develop new tools to support and accelerate research and LLM training.
  • Collaborate with engineering teams and scientific teams to implement experiments on cluster and data infrastructure.

Cohere is a security-first enterprise AI company building cutting-edge foundation AI models and end-to-end products for real-world business problems. The company is a global team of researchers, engineers, and designers passionate about AI, headquartered in Toronto with offices worldwide.

US UK Unlimited PTO

  • Design and optimize training and post-training pipelines for large language models.
  • Improve model quality through supervised fine-tuning, reinforcement learning, and evaluation.
  • Build PyTorch-based training infrastructure and optimize distributed training across multi-GPU environments.

Lightning AI builds an end-to-end platform for developing, training, and deploying AI systems, founded in 2019. They are a global company with offices in New York, San Francisco, Seattle, and London, backed by major venture capital firms, and foster a builder culture that values urgency, ownership, and open communication.

US

  • Design, train, and improve large-scale machine learning models for recommendation and personalization, leveraging modern deep learning architectures.
  • Own end-to-end delivery of major ML system components, from problem framing to production rollout, with cross-functional partners.
  • Optimize distributed training, model efficiency, and online inference performance while ensuring low-latency, high-throughput production systems.

Reddit is a community of communities built on shared interests, passion, and trust, hosting authentic conversations. With over 100,000 active communities and about 130 million daily active users, Reddit is one of the internet's largest sources of information.

Global 6w PTO 26w maternity 26w paternity

  • Design and build scalable inference pipelines running on large GPU clusters for synthetic data generation.
  • Conduct data ablations and experiments to enhance model performance and data quality.
  • Research and implement innovative synthetic data curation methods using Cohere's infrastructure.

Cohere is a leading security-first enterprise AI company that builds cutting-edge foundation AI models and end-to-end products for enterprises. They are a global team of researchers, engineers, and designers with offices in Toronto, London, New York, San Francisco, Montreal, Paris, Berlin, and Seoul.

North America

  • Use AI/ML expertise to solve real-world problems and develop new methods for LLM training and evaluation.
  • Research, propose, implement, train, and evaluate models and techniques end to end.
  • Build and maintain training pipelines and evaluation harnesses to ensure reproducibility and measurable progress.

ServiceNow is an AI platform company that helps organizations work smarter with AI and automation. It serves over 8,100 clients, including 85% of the Fortune 500, and fosters an AI-native culture.

Global 6w PTO 26w maternity 26w paternity

  • Build new RL environments targeting different agentic capabilities and industry areas.
  • Train and evaluate agents in those environments, improving both agents and environments.
  • Work across modeling and product to identify performance gaps and automate measurement.

Cohere is a security-first enterprise AI company building cutting-edge foundation models and end-to-end products. It is a global team of researchers, engineers, and designers headquartered in Toronto with offices worldwide.

$178,662–$213,680/yr
US

  • Build and scale ML infrastructure and tools to support AI-driven genetic medicine research.
  • Partner with scientists and engineers to translate research prototypes into robust systems.
  • Optimize model training and inference performance using advanced computing frameworks.

Dyno Therapeutics is building high-performance genetic technologies to transform patient lives. Their team unites world-class experts at the intersection of AI and genetic medicine.

Global 6w PTO 26w maternity 26w paternity

  • Build and ship features for North, Cohere's AI workspace platform.
  • Develop autonomous agents that interact with sensitive enterprise data.
  • Collaborate with researchers to productionize state-of-the-art models and techniques.

Cohere is a security-first enterprise AI company that builds cutting-edge foundation models and products for business. Headquartered in Toronto with offices globally, the company has a team of researchers, engineers, and designers passionate about their craft.

Europe

  • Develop and ship new features for Junie across CLI and model integrations.
  • Experiment with model setups, ensembling, and cost efficiency strategies.
  • Build and test autonomous agent behaviors for long-term project maintenance.

We create intelligent software development tools for developers and teams. Over 15 million users and 300,000 companies rely on our products, and our mission is to make development teams more productive and AI adoptable at scale.

Unlimited PTO

  • Build the environments and verifiers our models train against
  • Own the synthetic data pipeline, from generation through quality gates
  • Ship models into production and keep improving them on real usage

LiveKit builds the infrastructure layer for the voice-driven era of computing, enabling developers to build, deploy, and scale voice AI applications. Founded in 2021, they power voice AI for major companies and have a small, senior team that values craft and creativity.

$149,000–$219,000/yr
US Unlimited PTO

  • Build and maintain production-quality reference implementations and educational content for developers.
  • Represent the company at conferences, on livestreams, and within developer communities.
  • Work closely with product and engineering teams to identify friction and improve developer experience.

The company develops modern orchestration and AI technologies. It is a remote-first, high-performance environment with a focus on autonomy and developer community.

Global 6w PTO 26w maternity 26w paternity

  • Work directly on petabyte-scale storage infrastructure, and the networking and performance challenges that come with it.
  • Collaborate daily with researchers and engineers who are some of the best in the world at what they do.
  • Build and maintain the high-performance data layer that Modeling teams rely on for training and evaluation jobs.

Cohere is a security-first enterprise AI company that builds cutting-edge foundation AI models and end-to-end products. They are a global team of researchers, engineers, and designers passionate about their craft, with offices in Toronto, London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul.

$100,000–$300,000/yr
US

  • Build the software backbone for autonomous foundation models, including multimodal data pipelines and training workflows.
  • Implement and iterate on LLM, VLM, and VLA architectures, owning model code paths and inference runners.
  • Deliver production-grade serving tooling for low-latency operation and own systems from architecture to iteration.

We're building the next generation of ground transportation with advanced physical AI to simplify freight challenges. Our stealth team, founded by engineers who scaled autonomous driving, is developing a new vehicle platform and focuses on creating reliable real-time autonomous systems.

$120–$120/hr
Global

  • Design and implement tool-based environments for knowledge work simulations.
  • Write clean and modular code to support reinforcement learning training pipelines.
  • Troubleshoot and refine environment mechanics based on testing feedback.

Terac is building the world's largest pool of vetted human experts for AI. They are a platform for researchers and AI labs to recruit, screen, and pay study participants across industries and skill sets.

North America Unlimited PTO

  • Build reference architectures, benchmarks, and documentation that engineers trust.
  • Own the developer community, answering hard questions and setting the tone.
  • Write production-grade code for integrations and tooling that lower the barrier for new users.

Andromeda Cluster gives early-stage startups access to scaled AI infrastructure once reserved for hyperscalers. We are a unicorn with a small senior team, building the liquidity layer for global AI compute.

Global 6w PTO 26w maternity 26w paternity

  • Design, build, ship, and maintain core capabilities for North’s Agents & Automations platform.
  • Build product and platform features that help users create, run, debug, evaluate, and improve agents and automations.
  • Own features end-to-end, from technical design through implementation, testing, launch, and iteration.

Cohere is a security-first enterprise AI company that builds cutting-edge foundation models and end-to-end products for enterprises. They are a global team of researchers, engineers, and designers headquartered in Toronto with offices worldwide.

$150,000–$225,000/yr
US Canada 6w PTO 26w maternity 26w paternity

  • Build new integrations on the North agentic platform and maintain existing ones.
  • Own the integration pipeline from design to shipping to ongoing maintenance.
  • Help other engineers add integrations and create evals and tests for reliability.

Cohere is the leading security-first enterprise AI company building cutting-edge foundation AI models and end-to-end products for enterprises. A global technology team headquartered in Toronto with offices in London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul, they are passionate about driving AI adoption.

US

  • Architect and lead the development of next-generation, large-scale machine learning techniques.
  • Define and execute the ML strategy, identifying opportunities to enhance personalization and recommendation quality across Reddit.
  • Partner with ML infrastructure teams to build high-performance, distributed training systems that efficiently scale across multiple GPUs and cloud environments.

Reddit is a community of communities, built on shared interests, passion, and trust. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information.

India

  • Build and improve products and infrastructure for AI-powered teams.
  • Contribute across multiple product areas, including CI infrastructure and agentic applications.
  • Take ownership of projects from concept through delivery, with a focus on usability and quality.

Our partner is an early-stage engineering team building infrastructure and tools that help AI-powered teams work more efficiently. It is a high-growth, profitable startup with a small, high-impact team.

Global 7w PTO

  • Own the ML strategy for dialogue systems, lead a team of 3 ML engineers, and drive LLM post-training and model adaptation.
  • Build evaluation layers, cut dialogue failure modes, and keep inference efficient on latency and cost.
  • Stay hands-on with prototypes, debugging agent traces, and reviewing team work.

Social Discovery Group (SDG) is a group of social discovery companies that solve problems of loneliness and isolation through social entertainment platforms. The company has an international team of digital nomads and has been recognized as a Great Place to Work winner and a top company for remote jobs.