Diffuse Bio Inc. Logo

Diffuse Bio Inc.

Machine Learning Research Engineer

Posted Yesterday
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

6 Hours Ago
In-Office
Palo Alto, CA, USA
140K-280K Annually
Mid level
140K-280K Annually
Mid level
3D Printing • Consulting • Design • Manufacturing
Build scalable agentic frameworks and reproducible experimental pipelines that integrate frontier multimodal models with rich neuroscience data. Implement orchestration, MLOps, and cloud infrastructure, balancing research and production engineering on a small collaborative team.
Top Skills: Agent Orchestration Frameworks (Claude Code Agent TeamsAgentic FrameworksAzure)BeadsClaude Computer UseCodexCodex Multi-AgentsCrew Ai Agent Teams)Fdm-1GCPGemini-CliManusMicrosoft Agent FrameworkMlflowMultimodal Large Language ModelsOai OperatorOpenspecOrchestration Platforms (AwsPrompt EngineeringPythonPyTorchQwen Code)TransformersTui/Cli Coding Agents (Claude CodeWeights & Biases
6 Hours Ago
In-Office
Palo Alto, CA, USA
175K-250K Annually
Mid level
175K-250K Annually
Mid level
3D Printing • Consulting • Design • Manufacturing
Design and build an end-to-end, high-throughput dataloading stack for massive multimodal datasets: formatting, preprocessing, filtering, sharding, caching, and streaming data to distributed GPU training with observability, reliability, and performance benchmarking.
Top Skills: AirflowC++CudaDagsterDockerGpuKubernetesMlflowPrefectPythonPyTorchRustW&B
6 Hours Ago
In-Office
Palo Alto, CA, USA
200K-280K Annually
Mid level
200K-280K Annually
Mid level
3D Printing • Consulting • Design • Manufacturing
Optimize training and inference performance of foundation models by writing and tuning CUDA/Triton GPU kernels, profiling and removing bottlenecks, implementing low-precision and mixed-precision strategies, optimizing MoE routing and expert dispatch, integrating high-performance libraries, and collaborating closely with researchers to translate model ideas into scalable, efficient implementations.
Top Skills: CudaCudnnCutlassFlashattentionGpu KernelsKv-Cache ManagementLow-Precision TrainingMixed-Precision TrainingMoe (Mixture Of Experts)Moe RoutingNcclNsight ComputeNsight SystemsPost-Training QuantizationPyTorchPytorch ProfilerQuackSpeculative DecodingTensorrt-LlmTransformer ArchitecturesTritonVllm

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