Diffuse Bio Inc. Logo

Diffuse Bio Inc.

Machine Learning Research Engineer

Reposted 5 Hours Ago
In-Office
Redwood City, CA, USA
Mid level
In-Office
Redwood City, CA, USA
Mid level
Extend and scale in-house deep generative modeling tools for molecular design; run deep learning experiments to improve models or add functionality; and collaborate with software engineers to build efficient training and deployment systems for AI-driven protein therapeutics.
The summary above was generated by AI

At Diffuse Bio we’re building generative AI for protein therapeutics. Our team has been behind breakthroughs in AI protein design for the past 6 years, including the first experimental validation of AI-generated proteins and diffusion models for protein structure and sequence. We’ve built the first entirely AI computational nanobody design platform and are testing designs in lab now. 

Our goal is to build AI systems that can design protein therapeutics for the most challenging and high-value targets. We see enormous potential for AI to revolutionize drug discovery and expand the set of diseases we can treat. 

We are looking for candidates who are excited about the opportunity to join the founding team and play an expanding role in the company.


The role:

  • Extend and scale Diffuse’s in-house deep generative modeling toolkit for downstream applications in molecular design.

  • Thoughtfully execute deep learning experiments to improve performance of models or develop new functionality (e.g. loop engineering, structure prediction of protein-protein complexes).

  • Work closely with software engineers to build systems for efficient training and deployment of deep learning models.

Ideal background:

  • Self-starter who enjoys working on tough scientific problems and is results-driven.

  • Able to think critically, methodically, and creatively about experiments.

  • Proficient in Python.

  • Experience working with deep learning frameworks (e.g., PyTorch).

  • 3+ years of industry experience in a data science or engineering position.

  • Track record of impressive work in industry/academia centered on ML / deep learning.

  • Graduate degree in math, CS, stats, bioengineering, comp bio, or a related field (not a hard requirement for exceptional candidates).

  • Is located in the Bay Area (remote work is an option for exceptional candidates).

Pluses:

  • Knowledge of physics, math, molecular biology, chemistry, etc.

  • Previous work on ML applied to problems in structural biology or molecular design.

  • Strong publication record.

What we offer:

  • The opportunity to join the founding team and play a critical and expanding role in shaping the company.

  • The opportunity to work on cutting-edge AI with leading researchers from top institutions.

Similar Jobs

3 Days Ago
In-Office
San Francisco, CA, USA
Mid level
Mid level
Artificial Intelligence • Biotech
Build and optimize distributed machine learning infrastructure for large-scale model training and inference. Responsibilities include profiling training runs, improving compute, communication, and storage efficiency, developing parallelism and quantization strategies, creating CUDA and Triton kernels, and ensuring fault tolerance, checkpointing, deterministic orchestration, and reliability across GPU clusters. The role partners closely with research scientists to scale new model architectures and training recipes.
Top Skills: CudaDistributed SystemsGpu ClustersJaxPythonPyTorchTriton
11 Days Ago
Remote or Hybrid
San Francisco, CA, USA
165K-310K Annually
Entry level
165K-310K Annually
Entry level
Artificial Intelligence • Machine Learning • Software
Develop and post-train deep learning models while building systems, tooling, and workflows for training, evaluation, debugging, and deployment. Contribute to open-source projects, distributed AI infrastructure, backend services, and developer platforms. Collaborate with research, product, infrastructure, and customer teams to solve complex technical problems, prototype ideas, and productionize successful experiments.
Top Skills: Cloud InfrastructureCudaDeepspeedDistributed SystemsFsdpHugging FaceLightning AiNvidia MoltPyTorchSglangTritonVllm
12 Days Ago
In-Office
San Francisco, CA, USA
Senior level
Senior level
Artificial Intelligence • Logistics • Machine Learning • Software
Develop and productionize machine learning models for supply chain forecasting and optimization. Responsibilities include creating model architectures, curating proprietary real-world data, integrating LLM reasoning and tool use, designing evaluations, and continuously improving performance as market conditions change. The role owns the full research cycle from hypothesis through production and requires collaboration in a fast-moving, in-person environment.
Top Skills: Large Language Models (Llms)Machine LearningMathematical ModelingOptimizationTime-Series Forecasting

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account