Senior Machine Learning Engineer (Infra)

Genesis Therapeutics Inc.
Burlingame, California, US
Full-time

You Will

  • Lead engineering efforts focused on continuous improvement of the AI platform, focused on rapid build out and iteration on scalable and robust distributed infrastructure for ML training, inference, and evaluation.
  • Support model training and deployment across multiple clusters and multiple clouds, optimizing for throughput and cost.
  • Optimizing efficiency of ML models and other workloads in terms of latency, throughput, memory consumption, etc. (e.g., via GPU performance engineering), pushing the limits of what’s possible with the current hardware.
  • Define the long-term vision for Genesis’ ML platform.
  • Have the opportunity to mentor and guide more junior members of our technical team as well as research interns, fostering an environment of growth and innovation.

You are

  • Strong engineer who constantly strives for technical excellence. You can write clean code and have a deep understanding of the codebases you work in.
  • Deeply experienced with distributed training and inference of large models on GPU clusters and some of the core libraries and frameworks we use : Pytorch, Pytorch Lightning, Pytorch Geometric, and Ray.
  • Independent thinker with a strong sense of ownership and capability of engineering robust systems from first-principles-based conceptualization to state-of-the-art realization.
  • Curious, problem-oriented thinker who is excited to dive deep into the emerging field at the intersection of AI, physics, chemistry, and biology and make foundational contributions and discoveries (no previous experience in anything but ML necessary).

Nice to haves

  • Experienced with building, maintaining and debugging low-level cluster infrastructure running on multiple clouds using Kubernetes and Terraform.
  • Experienced GPU engineer who can quickly figure out performance bottlenecks and architect highly performant code for large scale ML workloads.
  • Experience with XLA, Triton, CUDA, or similar accelerator programming languages and / or deep learning compiler stacks.
  • Experience working with some of the following : molecular systems (protein sequences and 3D structures, small molecules, etc.

ML force fields or other physics-informed models and methods, or point cloud data in other application domains, such as 3D graphics.

Compensation, Benefits, and Perks

  • Competitive compensation package that includes salary and equity.
  • Comprehensive health benefits : Medical, Dental, and Vision (covered 100% for the employees).
  • 401(k) plan.
  • Open (unlimited) PTO policy.
  • Free lunches and dinners at our offices.
  • Paid family leave (maternity and paternity).
  • Life and long- and short-term disability insurance.

J-18808-Ljbffr

14 hours ago
Related jobs
Promoted
Genesis Therapeutics Inc.
Burlingame, California

Lead engineering efforts focused on continuous improvement of the AI platform, focused on rapid build out and iteration on scalable and robust distributed infrastructure for ML training, inference, and evaluation. GPU performance engineering), pushing the limits of what’s possible with the current h...

Promoted
Robinhood
Menlo Park, California

As a Senior ML Engineer, you will aid ML engineers in identifying and deploying critical models that bolster predictability across Robinhood businesses, products, and operations. Your expertise will be instrumental in crafting a highly reliable and scalable infrastructure, making the deployment of M...

Promoted
Pinterest
San Francisco, California

In this role, you will be responsible for developing and executing a vision for the evolution of the machine learning technology stack for the conversion modeling team. Lead the tech direction and responsible for the development of state-of-the-art applied machine learning projects for the ads conve...

Promoted
Coupang
Mountain View, California

We are seeking an experienced and passionate engineer leader who can drive this effort across the stack, including serving, pipelines, and machine learning. It will offer the autonomy to explore various areas and discover the next big innovation that infuses machine learning and deep learning into t...

Promoted
PlayStation
San Mateo, California

This role could suit an experienced embedded/games industry programmer with a passion for Machine Learning, or an accomplished Machine Learning/Vision programmer with a good understanding of computer architecture and performance. Machine Learning Engineer for Game Technology. We are looking for an e...

CoStar Group
CA, Orange County

Machine Learning Engineer, Data Scientist or related role. Collaborate on the continued improvement of CoStar’s cloud-based machine learning environment. Collaborate with other engineers, product owners, and leadership to create ML solutions that solve practical problems and improve our customer’s e...

Motion Recruitment Partners LLC
Mountain View, California

As a Machine Learning Engineer specializing in 3D Data Processing, you will be responsible for developing and implementing machine learning models that analyze and interpret complex 3D datasets. We are seeking a talented Machine Learning Engineer with expertise in 3D data processing to join our dyna...

NVIDIA
Remote, CA, US
Remote

NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to...

Truva
San Francisco, California

We are in search of a Senior LLM ML Engineer who is passionate about driving innovation in generative AI and machine learning. Advanced Knowledge in Machine Learning. Deep understanding of machine learning, NLP, and generative models, especially LLMs. Expertise in Python and familiarity with machine...

Quantcast
San Francisco, California

Machine Learning Engineer you care about the health and maintainability of our systems and the velocity of the engineering teams. The Modeling team is responsible for Machine Learning (ML) systems at Quantcast. Run Machine Learning experiments to test new modeling ideas. Keep up to date with develop...