Remote Software engineering Jobs · PyTorch

Job listings

$250,000–$295,000/yr
US Unlimited PTO 26w maternity 14w paternity

  • Define technical vision for agentic AI systems processing global trade data.
  • Lead development of production-grade ML pipelines from ingestion to inference.
  • Mentor engineers and collaborate cross-functionally to drive measurable impact.

This role is with a partner company of Jobgether building AI systems that transform global trade data into actionable intelligence for risk management and compliance. They offer a remote-friendly culture with flexible time off, comprehensive benefits, and opportunities for technical leadership.

  • Build and optimize RL infrastructure for large-scale training.
  • Improve efficiency across compute, memory, networking, and scheduling.
  • Design low-level optimizations, from kernels to runtime.

Prime Intellect builds the open superintelligence stack, providing full-stack infrastructure for frontier-scale AI post-training and deployment. The company has raised $150M from top investors and is mission-driven, focused on open-source AI and accelerating science.

  • Develop RL algorithms for reasoning, coding, and long-horizon tasks.
  • Design and run scaling experiments to improve training recipes across model sizes.
  • Extend RL training frameworks for asynchronous workflows and reliability.

Prime Intellect builds the open superintelligence stack, offering a platform that unifies compute, training, and deployment for frontier AI teams. With $150M in funding and a mission-driven culture, they focus on open-source AI research and real infrastructure.

$150,000–$350,000/yr

  • Build inference systems powering RL research, scaling toward 1–3T parameter models on NVL72 clusters.
  • Optimize rollout throughput, multi-node MoE inference, and weight-update overhead for long-horizon agents.
  • Collaborate with model researchers on architecture choices balancing model quality and execution cost.

Prime Intellect is building the open superintelligence stack, unifying compute, environments, and deployment for frontier-scale AI post-training. Backed by $150M in funding, the team is mission-driven and hard-working, focused on accelerating open AI research.

Global Unlimited PTO

  • Design, build, and evolve systems for distributed model training and batch inference at scale.
  • Develop workflows for experiment tracking, model evaluation, and registry management.
  • Optimize training and inference performance across GPU utilization, data loading, and distributed communication.

NOETIK is a frontier therapeutics company using AI to understand human biology and improve oncology outcomes, building foundation models on proprietary multimodal data. It fosters a collaborative, interdisciplinary culture focused on solving hard problems together.

Global 6w PTO 26w maternity 26w paternity

  • Design, run, and analyse large-scale RL post-training experiments such as RLHF and RL from verifiable rewards.
  • Research and develop methods that improve out-of-distribution generalization, including reward design, task diversity, and regularization.
  • Build held-out evaluations to measure generalization and catch reward hacking, and turn successful research into production training recipes.

Cohere is a leading security-first enterprise AI company building cutting-edge foundation models and end-to-end products for businesses. We are a global team of researchers, engineers, and designers passionate about their craft, headquartered in Toronto with offices worldwide.

  • Distributed pretraining: Implement and optimize model-parallel training on heterogeneous GPUs under low-bandwidth, high-latency links.
  • Performance optimization: Apply techniques that reduce communication overhead while maintaining model convergence in challenging network environments.
  • Elasticity and fault tolerance: Build robust checkpointing and recovery so runs survive node churn as participants join and leave.

Pluralis Research is a decentralized AI research company developing Protocol Learning to train large models on consumer devices across the internet. The team is world-class and deeply technical, backed by tier-1 investors, with a remote-first culture.

  • Build, train, and evaluate machine learning models end to end, from data and experiment design through training and reporting.
  • Design reinforcement learning environments, reward functions, and verifiers, and fine-tune language models using methods like PPO, DPO, and GRPO.
  • Collaborate with domain experts, analyze datasets, and present findings clearly to both technical and non-technical audiences.

EnableComp provides specialty revenue cycle management solutions for healthcare organizations using its unified E360 RCM platform. It is a multi-year Top Workplaces recipient and has been named to the Inc. 5000 list of fastest-growing private companies for eleven years.

  • Build the RL post-training stack end-to-end: rollout ingestion, reward computation, policy updates, and returning updated weights to the network.
  • Invent algorithms for asynchronous, high-latency, partially trusted generation, including staleness tolerance and off-policy corrections.
  • Ship the first decentralized post-trained models, building evals and taking the release from run to public artifact.

Pluralis Research is developing Protocol Learning, a decentralized approach to training and serving large models on consumer devices. We are a small, deeply technical team backed by Union Square Ventures, with a remote-first culture and team members spread globally.

  • Lead the end-to-end machine learning lifecycle, from data preparation and model training to evaluation, inference, and deployment.
  • Build and optimize scalable training, fine-tuning, and inference pipelines for large-scale models on GPU infrastructure.
  • Establish production best practices for model quality, reliability, safety, and measurable performance improvements.

This company builds production-grade machine learning systems and large-scale GPU-powered inference pipelines. It is a fully remote team that values technical rigor, collaboration, and hands-on engineering.