Taro Logo

Machine Learning Engineer- GPU

Bioptimus is a company focused on improving medical research through advanced machine learning and AI technologies.
Machine Learning
Senior Software Engineer
Remote
5+ years of experience
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Machine Learning Engineer- GPU

Bioptimus is seeking a Machine Learning Engineer- GPU to improve medical research using state-of-the-art machine learning algorithms. You'll work with interdisciplinary teams to build foundation models of biology for AI applications in biomedical innovation.

Key Responsibilities:

  • Implement and apply advanced machine learning methods, particularly in foundation models, representation learning, large language models, and generative AI.
  • Contribute to methodological research to unlock novel applications.
  • Craft high-performance CUDA kernels and manage GPU clusters.
  • Extract maximum performance from modern GPUs for efficient model inference and training.
  • Integrate low-level efficient code in high-level MLOps frameworks.

Requirements:

  • MSc or PhD in Computer Science or related field, or equivalent experience.
  • Expert in GPU programming and distributed computing.
  • Strong experience with deep learning frameworks (TensorFlow, PyTorch, Jax).
  • Practical expertise in representation learning, large language models, and generative AI.
  • Excellent Python coding skills and best practices.
  • Strong analytical and communication skills.

This role offers the opportunity to work at the intersection of healthcare/biology and AI, contributing to groundbreaking research in biomedical innovation. Join a dynamic team setting the foundations for the development and application of foundation models in biomedical research.

Location: London, Paris, Nantes, or Remote

Last updated a year ago

Responsibilities For Machine Learning Engineer- GPU

  • Implement and apply advanced machine learning methods in foundation models, representation learning, large language models, and generative AI
  • Contribute to methodological research in key areas
  • Drive the implementation of machine learning methods to relevant internal and external projects
  • Craft high-performance CUDA kernels
  • Extract maximum performance from modern GPUs
  • Integrate low-level efficient code in high-level MLOps frameworks

Requirements For Machine Learning Engineer- GPU

Python
  • Expert in crafting high-performance CUDA kernels
  • Deep knowledge of distributed computing frameworks for GPU clusters
  • Experience with modern GPUs (like H100) for efficient model inference and training
  • Understanding of generative AI and language models
  • Expertise in representation learning/large language models/generative AI
  • Strong analytical skills
  • Excellent command of coding in Python
  • MSc or PhD in Computer Science-related field or equivalent experience
  • Strong expertise with deep learning frameworks (Tensorflow, PyTorch, Jax, etc.)
  • Excellent written and oral communication skills

Interested in this job?