Source Job

Global

  • Design and implement novel training optimization techniques for large-scale neural networks.
  • Investigate approaches to improve training efficiency, stability, and convergence speed.
  • Collaborate with infrastructure and inference engineering teams to translate research into production performance.

Machine Learning Deep Learning Python PyTorch

11 jobs similar to AI Researcher — Training Optimization

Jobs ranked by similarity.

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.

Global

  • Design, develop, train, and evaluate machine learning models for interpreting complete CT studies.
  • Research foundation-model approaches for medical imaging, including 3D and volumetric learning at large scale.
  • Develop vision-language models connecting medical images with radiologist terminology and reporting patterns.

Our partner is a research-driven team building next-generation AI models to understand complete CT studies. They work with a dataset of approximately 10 million patients and collaborate with fellowship-trained radiologists across multiple specialties.

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.

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

US Unlimited PTO

  • Own a multimodal ML work-stream from problem definition through deployment, translating clinical needs into clear ML objectives and building systems with transformers, self-supervised learning, and segmentation.
  • Design rigorous evaluations beyond offline metrics, partner with engineering to productionize models, and investigate failure modes like laterality errors and hallucination.
  • Communicate research findings clearly, mentor less experienced researchers, and contribute to the research roadmap by identifying promising approaches.

Rad AI is an AI-driven healthcare company revolutionizing radiology by saving time, reducing burnout, and improving patient care. With over $140M in funding, a valuation of $528M, and partnerships with thousands of radiologists, Rad AI is a fast-growing team of mission-driven innovators.

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

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

Europe

  • Develop and scale latent video diffusion models for human-centric video generation.
  • Lead end-to-end applied research and engineering projects from hypothesis to production.
  • Optimize distributed training and inference for large-scale deployment.

They develop production-grade foundation models for human-centric video generation. The team operates in a high-ownership, fast-paced environment with a focus on safety and ethics.

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.

UK

  • Design, develop, and deploy AI models to solve complex business challenges.
  • Collaborate with cross-functional teams to integrate AI solutions into production systems.
  • Continuously improve and optimize AI pipelines for performance, scalability, and reliability.

We are passionate about helping businesses navigate the rapidly changing and complex world of emerging technologies. We create well-structured, secure, scalable solutions at speed to provide the foundation for groundbreaking change.

US

  • Develop and deploy deep learning models to predict therapeutic intervention effects.
  • Design experiments and collaborate with cross-functional teams including biologists and wet lab scientists.
  • Validate models and communicate results to diverse audiences to drive informed decision-making.

Absci is a clinical-stage biotechnology company advancing novel therapeutics using generative AI. It is a global company with offices in the US, Switzerland, and Serbia, fostering a collaborative and innovative culture.

$215,250–$285,982/yr
US

  • Build cutting-edge technology using deep learning and machine learning to personalize Pinterest.
  • Partner with teams to improve ML models for product surfaces like Homefeed and Search.
  • Use data-driven methods to improve content recommendation and ads delivery.

Pinterest is a platform where millions of people find creative ideas and plan for memories. The company is mission-driven, focusing on innovation and inclusion, with a large and diverse user base.