AI Researcher at Frontier Biology AI Lab jobs in United States
cer-icon
Apply on Employer Site
company-logo

Jack & Jill · 3 days ago

AI Researcher at Frontier Biology AI Lab

Jack & Jill is an AI recruiting firm representing a frontier biology AI research lab that applies machine learning to biological data. The role involves designing and implementing AI models for multi-omics, optimizing algorithms, and contributing to clinical outcomes in a multidisciplinary team.

Computer Software

Responsibilities

Implement and optimize high-performance ML algorithms for massive-scale training and reliable distributed computing
Design and maintain robust experimentation pipelines to enable rapid iteration and reproducible research outcomes
Refactor prototype research code into production-grade repositories with a focus on high-signal experimental design

Qualification

Large-scale foundation modelsPyTorchJAXLow-level optimizationCUDADistributed computingCollaborative researchHigh-signal experimental design

Required

Proven experience training, optimizing, and deploying large-scale foundation models (7B+ parameters) using PyTorch or JAX
Deep expertise in low-level optimization including CUDA, Triton, quantization, and hardware-specific tuning for GPUs or TPUs
Strong track record of solving hard technical problems and communicating complex approaches in collaborative research environments

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