Senior AI/ML Architect - Large Models, Remote & Equity

Absentia Labs; United States; Hybrid; USD 150,000 - 230,000; Full time

Benefits offered by this job

Equity participation

Flexible remote or hybrid work

Job summary

Absentia Labs is building intelligent systems at the intersection of AI, biology, chemistry, and engineering. We seek a Senior AI/ML Engineer to design, train, and deploy large-scale models that power our AI capabilities.

You will own training systems end-to-end, make architectural decisions, and lead cross-team collaboration. This role requires hands-on experience with LLMs, diffusion models, and GNNs, plus strong PyTorch and distributed training skills.

Qualifications

• 5+ years of industry experience in machine learning or applied AI roles.
• Experience training large-scale models in production settings.
• Hands-on expertise with LLMs, diffusion models, and/or GNNs.
• Strong proficiency in PyTorch (or equivalent frameworks).
• Deep understanding of distributed training and performance optimization.
• Experience working with large datasets and high-throughput pipelines.
• Strong software engineering fundamentals: clean code, testing, reproducibility, debugging at scale.
• Ability to communicate technical trade-offs to technical and non-technical stakeholders.

Responsibilities

• Design, train, and evaluate large-scale models, including LLMs, diffusion models, and GNNs.
• Own end-to-end training pipelines from dataset interfaces to distributed training and checkpointing.
• Make principled decisions about model architecture, objectives, optimization, and scaling laws.
• Build and optimize distributed training systems (data/model parallelism, sharding, mixed precision).
• Collaborate with data engineers to define ML-ready datasets and streaming interfaces.
• Translate ambiguous scientific or product requirements into robust ML solutions.
• Drive model evaluation, ablation, and iteration with focus on generalization and reproducibility.
• Contribute to architectural decisions around model serving and inference efficiency.
• Provide technical leadership through design reviews and mentorship.

Skills

PyTorch

LLMs

Diffusion models

GNNs

Distributed training

Model architectures

Job description

Absentia Labs is building intelligent systems at the intersection of AI, biology, chemistry, and engineering. We seek a Senior AI/ML Engineer to design, train, and deploy large-scale models that power our AI capabilities.

You will own training systems end-to-end, make architectural decisions, and lead cross-team collaboration. This role requires hands-on experience with LLMs, diffusion models, and GNNs, plus strong PyTorch and distributed training skills.