Skip to content
CohereCohereSan Francisco, CA

Senior Member of Technical Staff, Multimodal AI

Designs and develops cutting-edge multimodal AI systems integrating text, speech, and vision. Conducts research on representation learning using Python, JAX, PyTorch, TensorFlow, with expertise in distributed training and autoregressive models.

Salary not listed
RemoteAI Research

About the role

Responsibilities

  • Design and develop cutting-edge multimodal AI systems, integrating various modalities such as text, speech, and vision.
  • Conduct research and experiments on our advanced compute infrastructure, exploring novel ideas in multimodal representation learning, transfer learning, and more.
  • Collaborate closely with our world-class teams, learning from and contributing to their expertise in the field.

Requirements

  • Exceptional software engineering skills, with a proven track record of building robust and scalable systems.
  • Strong command of Python and popular deep learning frameworks like JAX, PyTorch, and TensorFlow, with understanding of their multimodal capabilities.
  • Knowledge of distributed training strategies, especially for large-scale multimodal models.
  • Familiarity with autoregressive models, particularly their application in multimodal tasks such as image or video captioning, speech-to-text generation.

Nice-to-Haves

  • Publications in top-tier venues demonstrating expertise in multimodal AI research.
  • Experience in writing efficient GPU kernels using CUDA, optimising performance for multimodal tasks.

Perks

  • Full health and dental benefits, including mental health budget.
  • 100% parental leave top-up for up to 6 months.
  • 6 weeks of vacation (30 working days).
  • Remote-flexible with co-working stipend.

Skills

PythonJAXPyTorchTensorFlowCUDADistributed TrainingAutoregressive ModelsMultimodal Ai

Similar roles

AI Research jobs
Modal

Member of Technical Staff

ModalNew York, NY +1

Conduct hands-on post-training research on LLMs including RL, distillation, and routing models. Collaborate with customers, labs, and engineering to turn techniques into production products and shape the research agenda. Requires proven research background in post-training LLMs and ability to ship impactful work.

150k – 350k/yr
On-siteAI Research
Ataraxis AI

Member of Technical Staff, Computational Pathology

Ataraxis AINew York, NY

Design, implement, and evaluate novel ML models for computational pathology to predict patient outcomes. Requires PhD in ML, computer vision or statistics, strong publication record, and expertise in PyTorch.

100k – 300k/yr
On-siteEntry levelAI Research
Ataraxis AI

Member of Technical Staff, Foundation Models

Ataraxis AINew York, NY

Design, implement, and evaluate novel self-supervised foundation models for clinical multi-modal data and precision medicine. Requires PhD in ML/statistics, strong publication record in top venues, and expertise in PyTorch/deep learning.

100k – 300k/yr
On-siteEntry levelAI Research
Ataraxis AI

Member of Technical Staff, Translational Science

Ataraxis AINew York, NY

Member of Technical Staff generating clinical insights from multi-modal AI models for precision medicine in oncology. Requires MD/PhD, lead authorship on high-impact papers, ML knowledge, and Python skills.

100k – 300k/yr
On-siteAI Research
Snowflake

Staff Applied Research Scientist

SnowflakeMenlo Park, CA

Lead applied research applying formal methods, automated reasoning, and AI (incl. LLMs) to improve correctness, reliability and velocity of Snowflake's cloud data platform and distributed systems. Translate ideas into production capabilities; requires PhD + 8+ years experience and track record of impact.

236k – 339k/yr
On-site8+ YOEAI Research