Machine Learning Research Engineer at Frontier AI biology lab jobs in United States
cer-icon
Apply on Employer Site
company-logo

Jack & Jill · 6 hours ago

Machine Learning Research Engineer at Frontier AI biology lab

Jack & Jill is recruiting for a Machine Learning Research Engineer at Frontier AI biology lab, where the candidate will design and scale foundational AI models for multi-omics at massive scale. This role focuses on implementing high-performance ML algorithms and maintaining robust experimentation pipelines to drive breakthroughs in scientific understanding, particularly in understanding and protecting the human brain through technology.

Computer Software

Responsibilities

Implement and optimize high-performance ML algorithms for reliability and massive scalability
Design and maintain experimentation pipelines for rapid iteration and reproducible research outcomes
Refactor prototype code into production-grade repositories and develop high-speed data processing workflows

Qualification

PyTorchJAXDistributed computingLarge-scale modelsAlgorithm optimizationHigh-performance ML algorithmsExperimentation pipelinesData processing workflows

Required

Deep expertise in modern ML frameworks like PyTorch or JAX and distributed computing principles
Proven experience training, optimizing, and deploying large-scale models with over 7B parameters
Skilled in low-level algorithm optimization including quantization, CUDA, or hardware-specific tuning

Company

Jack & Jill

twitter
company-logo
Ok, I'll go first. I'm Jack, an AI that gets to know you deeply, what you’re great at and what you want out of your career.

Funding

Current Stage
Early Stage
Company data provided by crunchbase