Senior/Staff Machine Learning Engineer - Perception Offline Driving Intelligence
As an engineer in the Offline Driving Intelligence (ODIN) team at Zoox, the responsibilities include developing advanced multimodal large language models to enhance environmental understanding for robotaxis, designing model architectures and training techniques using sensor inputs and large scale data, driving end-to-end machine learning solutions from research to production using Zoox's data pipelines and infrastructure, collaborating with perception, planning, safety, and systems teams to integrate models into the vehicle's decision-making pipeline, and validating and optimizing solutions using real-world driving scenarios to contribute directly to the safety and reliability of Zoox's autonomous system.
Senior Machine Learning Engineer
The Senior Machine Learning Engineer will research, evaluate, and implement state-of-the-art NLP methodologies and large language model approaches to drive product innovation and develop new functionalities. They will design, develop, and deploy LLM agents and multi-agent systems to automate complex legal workflows and enhance user experiences. The role involves collaborating on projects that leverage emerging technologies such as Retrieval-Augmented Generation (RAG) and Knowledge Graphs to enhance the core product and explore new use cases. The engineer will work closely with cross-functional teams to integrate advanced ML models and NLP solutions into the platform, ensuring alignment with business objectives and tangible value. Additionally, they will stay current with the latest trends and breakthroughs in NLP, machine learning, and multi-agent systems, contributing ideas to shape the strategic direction of AI initiatives.
Senior Machine Learning Engineer
Design and ship advanced ML systems, especially LLM-driven agents and self-improving workflows. Build robust data and training pipelines, enable fast experimentation, and ensure models and agents continuously improve in production. Build LLM-based agents, tool-using workflows, and autonomous self-improvement loops. Design, train, and evaluate ML models across NLP/LLM, vision, and retrieval domains. Develop data pipelines, training code, experiment tooling, and automated deployment systems. Use PyTorch for model development and W&B (or similar) for tracking experiments and lineage. Implement monitoring for performance, drift, safety, and agent behavior. Optimize inference for latency, throughput, and cost. Work closely with engineering and product teams to turn prototypes into reliable production features. Establish ML engineering best practices and mentor teammates.
Sr. Machine Learning Researcher
As a Senior Machine Learning Researcher at AKASA, you will lead the design, training, and evaluation of large language models to address healthcare-specific challenges, focusing on advancing clinical Natural Language Understanding. You will collaborate with cross-functional teams including PhD researchers, ML engineers, and healthcare experts to integrate Human-in-the-Loop data for model improvements and explore optimization methods. Your role includes working end-to-end on model design, data creation, training, evaluation, and iteration to ensure research advances both models and real-world healthcare tasks. You will stay updated on machine learning advancements to maintain AKASA's leadership in healthcare AI, partner with healthcare experts to align models with real-world needs, contribute to high-impact publications, and support the integration of your research into AKASA's product offerings used across healthcare systems.
Tech Lead, LLM & Generative AI (Full Remote - Ukraine)
The Tech Lead is responsible for architecting the system and mentoring a team of three engineers while spending significant time hands-on in the codebase using Python and PyTorch. They will own the core chat loop, optimizing context windows, memory/RAG retrieval, and inference latency to ensure a seamless real-time experience. They must drive the strategy for supervised fine-tuning (SFT), reinforcement learning with human feedback (RLHF/DPO), deciding when to prompt, fine-tune, or architect new retrieval augmented generation (RAG) pipelines. They manage the "Data Engine" overseeing sourcing, labeling, and cleaning datasets to improve model steerability and multicultural performance. Additionally, they design and train custom classifiers for high-precision moderation to detect and filter non-consensual or illegal content, moving beyond binary safe/unsafe flags to enable nuanced, context-aware moderation systems within an uncensored, NSFW environment.
Tech Lead, LLM & Generative AI (Full Remote - Slovenia)
Lead the LLM team of 3 engineers by acting as both architect and hands-on coder, writing production code in Python/PyTorch, and mentoring the team. Own and optimize the core chat loop, including context windows, memory/RAG retrieval, and inference latency to ensure a real-time user experience. Drive the strategy for supervised fine-tuning (SFT) and RLHF/DPO (Preference Optimization), deciding when to prompt, fine-tune, or design a new RAG pipeline. Manage the data engine responsible for sourcing, labeling, and cleaning datasets to improve model steerability and multicultural performance. Architect and build sophisticated, context-aware moderation classifiers and alignment strategies to detect and filter non-consensual or illegal content in an explicit environment, moving beyond binary safe/unsafe flags.
Tech Lead, LLM & Generative AI (Full Remote - Slovakia)
The Tech Lead will ship code and lead from the front by architecting the system and mentoring the team while spending significant time hands-on in the codebase using Python and PyTorch. They will own the core chat loop by optimizing context windows, memory/retrieval-augmented generation (RAG) retrieval, and inference latency to ensure a seamless, real-time experience. They will own the model lifecycle by driving strategy for supervised fine-tuning (SFT) and reinforcement learning with human feedback (RLHF/DPO), deciding when to prompt, fine-tune, and architect new RAG pipelines. They will manage the sourcing, labeling, and cleaning of diverse datasets to improve model steerability and multicultural performance. Additionally, they will architect high-precision moderation by designing and training custom classifiers to detect and filter non-consensual or illegal content in an explicit environment and create nuanced, context-aware moderation systems beyond binary safe/unsafe flags.
Tech Lead, LLM & Generative AI (Full Remote - Norway)
The Tech Lead will ship production code and lead the LLM team of 3 engineers by acting as both architect and mentor. Responsibilities include owning the core chat loop by optimizing context windows, memory/RAG retrieval, and inference latency for a real-time experience. The role involves driving strategy for supervised fine-tuning (SFT), RLHF/DPO preference optimization, managing data sourcing, labeling and cleaning to improve model steerability and multicultural performance. Additionally, they will architect high-precision moderation systems by designing and training custom classifiers to detect and filter non-consensual or illegal content in an explicit environment, moving beyond binary safe/unsafe flags to nuanced, context-aware moderation systems.
Tech Lead, LLM & Generative AI (Full Remote - Moldova)
Lead the LLM team, owning the architecture, training, and deployment of models powering the core product. Act as a player/coach by architecting the system, mentoring the team, and actively writing production code primarily in Python/PyTorch. Optimize the core chat loop focusing on context windows, memory/RAG retrieval, and inference latency to deliver a seamless real-time user experience. Drive the strategy for model lifecycle management including supervised fine-tuning (SFT), reinforcement learning with human feedback (RLHF), and direct preference optimization (DPO), deciding when to prompt, fine-tune, or architect new retrieval-augmented generation pipelines. Manage the data engine involving sourcing, labeling, and cleaning datasets to enhance model steerability and multicultural performance. Architect and build high-precision moderation systems by designing and training custom classifiers to detect and filter non-consensual or illegal content in an explicit environment, moving beyond binary safe/unsafe flags towards nuanced, context-aware moderation.
Tech Lead, LLM & Generative AI (Full Remote - Austria)
The Tech Lead will act as a player/coach, architecting the system and mentoring the team while spending significant time hands-on in the codebase (Python/PyTorch). They will own the core chat loop, optimizing context windows, memory/RAG retrieval, and inference latency to ensure a seamless, real-time experience. They will drive the strategy for supervised fine-tuning (SFT) and reinforcement learning with human feedback/preference optimization (RLHF/DPO), deciding when to prompt, fine-tune, or architect a new RAG pipeline. They will manage the data engine overseeing the sourcing, labeling, and cleaning of diverse datasets to improve model steerability and multicultural performance. Additionally, they will architect high-precision moderation by designing and training custom classifiers to detect and filter non-consensual or illegal content within an explicit environment and create nuanced, context-aware moderation systems beyond binary safe/unsafe flags.
Access all 4,256 remote & onsite AI jobs.
Frequently Asked Questions
Lorem ipsum dolor sit amet, consectetur adipiscing elit. Suspendisse varius enim in eros elementum tristique. Duis cursus, mi quis viverra ornare, eros dolor interdum nulla, ut commodo diam libero vitae erat. Aenean faucibus nibh et justo cursus id rutrum lorem imperdiet. Nunc ut sem vitae risus tristique posuere.