Remote Jobs RockRemote Jobs Rock

Senior AI/ML Engineer

๐Ÿ”ฅ 19 hours ago
Machine LearningLarge Language ModelsDiffusion ModelsGraph Neural Networks
Apply Now

๐Ÿ“œ Description

  • Design, train, and evaluate large-scale LLMs, diffusion models, and Graph Neural Networks.
  • Own end-to-end training pipelines, including dataset interfaces, batching, distributed training, and checkpointing.
  • Choose model architectures, objectives, optimization strategies, and scaling approaches.
  • Build and optimize distributed training systems using parallelism, sharding, and mixed precision.
  • Evaluate and iterate on models for generalization, stability, and reproducibility.
  • Guide decisions on model serving, inference efficiency, and lifecycle management; mentor and collaborate across teams.

๐Ÿ› ๏ธ Requirements

  • You are a senior ML engineer who thinks holistically about models as systems. You are comfortable operating under uncertainty, making trade-offs between compute, data, and performance, and owning outcomes from research through production.
  • You care deeply about training dynamics, failure modes, and scaling behavior, and you have the scars to prove it.
  • Experience with reinforcement learning, fine-tuning, or preference-based optimization (e.g., RLHF).
  • Familiarity with model compression, distillation, or inference optimization.
  • Experience deploying models in production inference systems.
  • Exposure to multimodal learning or foundation models.
  • Prior work in startups or fast-moving R&D environments.
  • Contributions to open-source ML frameworks or research codebases.

โœจ Benefits

  • Competitive compensation, including meaningful equity participation, allows you to share directly in the long-term success and growth of the company.
  • The opportunity to work on foundation-level ML systems applied to real scientific problems.
  • Ownership over model design and training strategy, not just implementation.
  • Close collaboration with data, infrastructure, and scientific teams.
  • High autonomy, low bureaucracy, and a culture that values technical depth.
  • Flexible remote or hybrid work arrangements.
Full job description

About Absentia Labs

Absentia Labs is building intelligent systems that sit at the intersection of AI, biology, chemistry, and large-scale engineering. Our goal is to translate complex scientific data into machine intelligence capable of reasoning, generalizing, and driving discovery.

Biomedical data is fragmented, noisy, and deeply interconnected. Turning it into a useful signal requires not only strong data foundations but also carefully designed learning systems that can scale across modalities, tasks, and uncertainty regimes. This role focuses on building and training those systems.

The Role

As a Senior AI/ML Engineer, you will lead the design, training, and deployment of large-scale machine learning models that form the core of Absentia Labsโ€™ AI capabilities. You will work at the boundary between model architecture, training systems, and production infrastructure, with significant ownership over technical direction.

This role is intended for engineers who have trained large models in real production environments, understand the realities of scale, and can reason about both learning dynamics and systems constraints.

What Youโ€™ll Do

  • Design, train, and evaluate large-scale models, including Large Language Models (LLMs), diffusion models, and Graph Neural Networks (GNNs).

  • Own end-to-end training pipelines, from dataset interfaces and batching strategies to distributed training and checkpointing.

  • Make principled decisions about model architecture, objective functions, optimization strategies, and scaling laws.

  • Build and optimize distributed training systems (data parallelism, model parallelism, sharding, mixed precision).

  • Collaborate closely 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 a focus on generalization, stability, and reproducibility.

  • Contribute to architectural decisions around model serving, inference efficiency, and lifecycle management.

  • Provide technical leadership through design reviews, mentorship, and cross-team collaboration.

Who You Are

You are a senior ML engineer who thinks holistically about models as systems. You are comfortable operating under uncertainty, making trade-offs between compute, data, and performance, and owning outcomes from research through production.

You care deeply about training dynamics, failure modes, and scaling behavior, and you have the scars to prove it.

You Likely Have

  • 5+ years of industry experience in machine learning or applied AI roles.

  • Demonstrated experience training large-scale models in production settings, not just prototypes.

  • Hands-on expertise with LLMs, diffusion models, and/or GNNs.

  • Strong proficiency in PyTorch (or equivalent deep learning frameworks).

  • Deep understanding of distributed training, including parallelism strategies and performance optimization.

  • Experience working with large datasets and high-throughput data pipelines.

  • Strong software engineering fundamentals: clean code, testing, reproducibility, and debugging at scale.

  • Ability to clearly communicate technical trade-offs to both technical and non-technical stakeholders.

Bonus If You Have

  • Experience with reinforcement learning, fine-tuning, or preference-based optimization (e.g., RLHF).

  • Familiarity with model compression, distillation, or inference optimization.

  • Experience deploying models in production inference systems.

  • Exposure to multimodal learning or foundation models.

  • Prior work in startups or fast-moving R&D environments.

  • Contributions to open-source ML frameworks or research codebases.

Note: Prior experience with molecular or biomedical models is not required. We value strong ML systems experience and the ability to transfer learning across domains.

What We Offer

  • Competitive compensation, including meaningful equity participation, allows you to share directly in the long-term success and growth of the company.

  • The opportunity to work on foundation-level ML systems applied to real scientific problems.

  • Ownership over model design and training strategy, not just implementation.

  • Close collaboration with data, infrastructure, and scientific teams.

  • High autonomy, low bureaucracy, and a culture that values technical depth.

  • Flexible remote or hybrid work arrangements.

How to Apply

Please submit your resume and a brief note describing your experience training large-scale models. Links to GitHub repositories, papers, or technical write-ups are encouraged.

Our Commitment

Absentia Labs is an equal opportunity employer. We believe diverse teams build better systems and stronger science, and we encourage applicants from all backgrounds to apply.

Build and operate production conversational AI for phone and SMS, taking ML and LLM features from prototype to deployment while developing agent tooling, evaluation systems, and shared infrastructure.

PythonConversational AIMachine LearningLarge Language Models
Wiser Solutions

Senior Machine Learning Engineer

๐Ÿ”ฅ 21 hours ago
Wiser Solutions๐Ÿ‘ฅ 201 - 500 employees๐Ÿข Computer Software

Build and operate customer-facing machine learning systems, taking problems from rapid experimentation through production while balancing model quality, inference cost, and reliability at scale.

PythonMachine LearningLLMsRAG
Bloomreach

Senior Machine Learning Engineer

๐Ÿ•’ 3 days ago
Bloomreach๐Ÿ‘ฅ 1001 - 5000 employees๐Ÿข Internet

Build and operate production machine learning systems for large-scale search, from query-time models and catalog pipelines to model serving, evaluation, and reliable low-latency deployment.

PythonGoJavaMachine Learning
Instacart

Senior Machine Learning Engineer, Fulfillment

๐Ÿ•’ 7 days ago
Instacart๐Ÿ‘ฅ 10,000+ employees๐Ÿข E-commerce

Design, deploy, and operate production machine learning systems for Enterprise AI, owning the full model lifecycle and partnering across teams to deliver scalable solutions with measurable business impact.

Machine LearningPythonSQLFeature Engineering
Instacart

Senior Machine Learning Engineer, Fulfillment

๐Ÿ•’ 7 days ago
Instacart๐Ÿ‘ฅ 10,000+ employees๐Ÿข E-commerce

Design, deploy, and operate production machine learning systems for Enterprise AI products, owning the full lifecycle from problem formulation and experimentation to monitoring and continuous improvement.

Machine LearningPythonSQLFeature Engineering

Trusted by Remote Workers