Machine Learning Engineer

Senior ML Engineer - Computational Biology

Stealth AI-First Biotech | Tokyo/Remote | Build the Future of AI and Biology

Join a stealth, well-funded startup revolutionizing pharmaceutical research through cutting-edge AI.

Technology Stack: Neural ODEs/PDEs, Physics-Informed Neural Networks, Multi-Agent AI Systems, Distributed ML Infrastructure, Natural Language Interfaces for Scientific Computing.

What You'll Own

🧬 AI-Biology Integration

• Design neural differential equation frameworks to model biological systems at scale
• Build physics-informed neural networks for pharmacokinetic/pharmacodynamic simulation
• Integrate molecular dynamics with deep learning for drug-target interaction prediction

🤖 Autonomous AI Systems

• Develop AI agents that reason over scientific literature and experimental data
• Create retrieval-augmented systems for hypothesis generation and experimental design
• Build conversational interfaces that make complex biophysical modeling accessible to researchers

⚡ Production ML Infrastructure

• Deploy distributed training systems across GPU/TPU clusters for large-scale biological models
• Implement MLOps pipelines for continuous model improvement and A/B testing
• Scale inference systems to handle real-time molecular simulation queries

Ideal Background

Must Have:

• 4+ years production ML experience with PyTorch/JAX
• Strong mathematical foundation (differential equations, numerical methods, optimization)
• Experience with biological/chemical datasets OR strong willingness to learn domain quickly
• Track record building novel architectures from research papers to production

Highly Valued:

• Neural ODE/PDE implementation experience or physics-informed ML
• AI agent development (LangChain, LangGraph, autonomous reasoning systems)
• Computational biology, drug discovery, or scientific computing background
• PhD in quantitative field (Applied Math, Physics, Bioinformatics, Computer Science)

Mindset:

• Excited by unsolved technical challenges at the frontier of AI and biology
• Comfortable with ambiguity and rapid iteration in a research-heavy environment
• Passionate about technology that could fundamentally extend human healthspan

Why This Opportunity

• Lead technical architecture decisions for breakthrough biotech platform
• Substantial equity package with liquidation opportunities at funding milestones
• Clear path to VP Engineering/CTO role as company scales
• Competitive package
• Tokyo HQ with full relocation support (visa sponsorship, housing assistance)
• Remote options for exceptional candidates (US/Canada/EU timezones)
• Quarterly team gatherings in Tokyo + major AI conferences

🔬 Technical Excellence

• Work alongside world-class computational biologists and AI researchers
• Access to cutting-edge compute resources and proprietary biological datasets
• Collaborate with top-tier academic institutions and pharmaceutical partners