The Team

Our engineering team builds systems that interact with some of the most complex software  ever deployed in production. The team is composed of industry veterans that have built deep learning infrastructure, autonomous drones, ridesharing marketplaces, ad tech and much more. 

We are looking for a client-obsessed AI Solutions Engineer with entrepreneurial tendencies to join the good fight and help build out our Solutions Engineering org. You’ll be the trusted technical advisors for our customers, driving business value, offering advice, and growing accounts. You’ll accomplish this by leading customers to solutions oftentimes by teaching the product to new users or consulting on best practices. You must be ready for technical discussions with data scientists and engineers, then demonstrate the value of Arize in business discussions with directors and executives. The goal is to enable our customers to become successful and enthusiastic about Arize. While we are a remote-first company, the nature of this role requires candidates to be based in the San Francisco Bay area/PST time zone.

What You’ll Do

• Work closely with some of the most sophisticated ML / GenAI teams in the world.
  • You will act as a trusted advisor to our customers, while also building relationships with technical and business stakeholders.
• Advise on GenAI and ML best practices 
• Give ML and LLM product demos to technical and business stakeholders
• Run strategic business reviews for customers in partnership with our sales team
• Interface with our pre-sales engineering team to gather client goals and KPI’s.
• Partner with our product and engineering teams to help drive the product roadmap
• Spearhead new opportunities within existing accounts to help drive expansions.

What We’re Looking For

Note: Even if you do not check every single box, we still encourage you to apply! 

• Previous experience working as a Data Scientist, Machine Learning Engineer, or as an Engineer working with ML models or GenAI applications in production.
• Comfortable working in public Cloud environments (AWS, Azure, GCP)
• Knowledge of machine learning frameworks such as TensorFlow, PyTorch or Scikit-learn
• Knowledge of LLM / Agentic frameworks such as Llamaindex, LangGraph, and DSPy
• Understanding of ML/DS concepts, model evaluation strategies and lifecycle (feature generation, model training, model deployment, batch and real time scoring via REST APIs) and engineering considerations
• Understanding of GenAI concepts and application evaluation + development lifecycle  
• Proficiency in a programming language (Python, JS/TS, Java, Go, etc)
• Strong Communication Skills - Ability to simplify complex, technical concepts.
• A quick and self learner - undaunted by technical complexity of production ML deployments and welcome the challenge to learn about them and develop your own POV.

Bonus Points, But Not Required

• Previous engineering experience in:
  • Data Science
  • MLOps
  • ML Frameworks
  • LLM / Agentic frameworks
• Customer facing experience strongly preferred such as Solutions Architect, Implementation Specialist, Sales Engineer, Customer Success Engineer, Consultant, or Professional Service roles
• Prior experience working with applications deployed with Kubernetes
• Prior experience demoing technical products to both business and technical audiences

The estimated annual salary and variable compensation for this role is between $125,000 - $175,000, plus a competitive equity package. Actual compensation is determined based upon a variety of job related factors that may include: transferable work experience, skill sets, and qualifications. Total compensation also includes a comprehensive benefit package, including: medical, dental, vision, 401(k) plan, unlimited paid time off, generous parental leave plan, and others for mental and wellness support.

While we are a remote-first company, we have opened offices in New York City and the San Francisco Bay Area, as an option for those in those cities who wish to work in-person. For all other employees, there is a WFH monthly stipend to pay for co-working spaces.

About this role

As a Staff Machine Learning Platform Engineer, you will help design, improve, and operate a scalable ML platform to accelerate model training, deployment, and governance. You are the technical bridge between data science and production engineering.  You’ll be joining a small but deeply critical team that scales Faire’s ability to support tens of thousands of local businesses in a constantly narrowing retail landscape.

What You Will Do

• Design and operate ML infrastructure, including workspaces, clusters, jobs, and workflows
• Productionize ML workloads using Spark, Delta Lake, MLflow, and Databricks Workflows
• Teach data scientists how to utilize our ML platform to advance development from notebook to production for our most critical models
• Implement Unity Catalog for data governance, lineage, access control, and secure multi-tenant usage
• Build CI/CD pipelines for ML using Terraform and Git-based workflows (e.g., GitHub Actions)
• Optimize performance, reliability, and cost across training and inference workloads
• Configure Identity and Access Management (IAM) and Role Based Authentication Controls (RBAC) for sensitive data sets
• Establish observability for data quality, model performance, and platform health
• Build and maintain ML Platform technical documentation

What it takes

• 8+ years of experience building production ML or data platforms
• A degree (preferably graduate level) in Computer Science, Engineering, Statistics, or a related technical field
• Strong hands-on expertise with Databricks, Spark, Delta Lake, and MLflow.
• Proficiency in Python, SQL, and distributed systems concepts
• Experience with cloud platforms and infrastructure-as-code
• Solid understanding of MLOps best practices: CI/CD, monitoring, reproducibility, and security
• Experience supporting multiple ML teams in a shared platform environment
• Are an active owner of orphaned problems and are willing to assimilate whatever knowledge you’re missing to get the job done

Tech Stack

Faire uses a modern cloud based tech stack.  For this role, you’ll want to be proficient with the following:

Salary Range

Canada: the pay range for this role is $216,000 to $297,000 per year. 

This role will also be eligible for equity and benefits. Actual base pay will be determined based on permissible factors such as transferable skills, work experience, market demands, and primary work location. The base pay range provided is subject to change and may be modified in the future.

Faire uses Artificial Intelligence (AI) to screen and select applicants for this position.

This job posting is for an existing vacancy.

About Wholesail ⛵️

Wholesale trade, a $55T global market, still operates like the world before the internet and modern payment networks: collection processes are manual and run on paper, short-term credit risk is assumed by the vendor, and businesses rely heavily on legacy ERP systems that don't talk to each other. The result is large, measurable drag — we estimate that vendors spend over $500B a year globally across industries on an inefficient mix of processing, bad debt, lending, credit insurance, software, and labor costs to get paid.

Wholesail is building a financial network from the ground up that connects the systems of vendors and buyers involved in wholesale trade to enable streamlined payment and the transfer of risk to third parties. This will allow vendors to offload risk and eliminate tens of billions of waste — while giving creditworthy buyers better terms and third-party capital, unlocking hundreds of billions (ultimately trillions) in additional sales. The primitives of this network scale across industries and geographies: a universal approach to ERP integrations, modern payment rails, and a live trade-credit bureau to underwrite risk — which we're calling Lighthouse.

The Opportunity: Risk & Capital Products 🚢

Credit is the load-bearing beam of our network. Every time a vendor ships goods before getting paid, someone is taking a risk — today it's the vendor, tomorrow it should be a third party at a fair price. Getting that transfer right is what unlocks the next order of magnitude of sales across the wholesale economy, and the only way to get it right is to underwrite buyers more accurately than anyone else in the industry. Listen to the Visa episode of the Acquired podcast to learn how credit card networks did this for retail trade. 

We think we're uniquely positioned to do this. Through Lighthouse, we're building a live, reciprocal trade-credit bureau: vendors on our network contribute real-time payment behavior on a long tail of SMB buyers that no traditional bureau sees. That data — combined with the bank, ERP, and transaction signals already flowing through Wholesail — is a modeling dataset that doesn't exist anywhere else. The first MLE on this team gets to decide what we build with it.

The problems are real and the stakes are significant. Our models directly shape the terms buyers are offered and the losses Wholesail and our capital partners absorb. There's no established playbook here and no legacy stack to inherit — you'll be setting the direction for how we do modeling, data engineering, and production ML at Wholesail for years to come.

The Role ⚒️

As the founding MLE on the Risk Engineering & Capital Products team, you'll own credit risk modeling end-to-end — from problem framing and data pipeline design, through model development and validation, through production serving and monitoring. You'll work directly with the hiring manager, our engineering and product leadership, and our hands-on advisor Dan Massoni (former Chief Commercial Credit Officer of American Express) to decide what to build, explore features to model and how you’ll scale our approach risk scoring. .

Expect your work to span:

• Modeling. Design, build, and validate credit risk models (PD, LGD, EAD, fraud, exposure sizing, pricing) against our proprietary reciprocal-bureau data and external signals. Choose the right tool for the job — gradient-boosted trees, deep models, LLMs, or classical statistical models — and defend the choice.
• Data engineering. Build the pipelines, feature logic, and training datasets that your models (and future models) run on. At this stage, the MLE owns the full data path; over time we'll grow a team around you.
• Productionization. Ship models into production systems where they make real decisions on real dollars. Own deployment, monitoring, drift detection, and iteration.
• Partnering cross-functionally. Work closely with product, engineering, capital markets, and external lending and insurance partners. Explain your models to people who care about the outcomes but not the math, and translate their constraints back into modeling decisions.
• Shaping the team. Help define the hiring bar and interview process for the data scientists and MLEs who will join after you. You'll have a strong voice in what this team becomes.

Qualifications

• 5+ years of experience building models for production use cases.
• Deep modeling skill: strong command of supervised learning on tabular data, including feature engineering, model selection, hyperparameter tuning, calibration, and rigorous offline and online evaluation.
• Strong data engineering skills — comfortable owning ETL and feature pipelines end-to-end against real, messy production data (SQL and a modern data-processing stack).
• Proficiency in Python and the modern ML ecosystem (pandas, scikit-learn, PyTorch or TensorFlow, XGBoost/LightGBM, Jupyter, etc.).
• Solid statistical reasoning — you can spot leakage, selection bias, label noise, and spurious correlations, and you know why your offline metric doesn't always predict online performance.
• Track record of owning models end-to-end: not just prototyping in a notebook, but taking something from a problem statement to a production system that makes decisions.
• Excellent written and spoken English communication skills; ability to explain model behavior, tradeoffs, and limitations to non-ML audiences (product, ops, capital partners, leadership).
• The spirit of a team player who believes in fostering a healthy and supportive work environment.

Bonus Qualifications

• Direct experience in credit risk modeling — PD/LGD/EAD models, scorecards, underwriting models, exposure management, or collections modeling — particularly for SMB or subprime segments.
• Experience in fintech, lending, payments, fraud, or insurance, or with regulated modeling environments (model risk management, adverse action, fair lending).
• Experience building or operating ML platform components — feature stores, model registries, training orchestration, online serving, monitoring/drift detection.
• Backend engineering depth: comfortable writing production services, designing APIs, and reasoning about distributed systems that consume your models.
• Strong data analysis background — fluency in exploratory analysis, experimentation, and translating data into product and business decisions.
• Experience with LLMs and agentic systems applied to operational problems (document understanding, KYB, entity resolution, agent tooling).
• Experience as the first or founding ML hire at a company, or early on a team that had to build ML infrastructure from scratch.
• Published research, Kaggle placements, or prominent OSS contributions in the ML or data ecosystem.
• BA or BS in Computer Science, Statistics, Mathematics, a related technical field, or equivalent practical experience.
• Advanced degree (MS or PhD) in a quantitative field.

Equal Employment Opportunity

Wholesail is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. We do not discriminate on the basis of race, color, religion, national origin, sex, gender identity or expression, sexual orientation, age, marital status, veteran status, disability status, pregnancy, or any other characteristic protected by federal, state, or local law. Employment decisions at Wholesail are based on qualifications, merit, and business needs. We encourage applicants from all backgrounds to apply.

• Help design, train, with a focus towards deploying onboard machine learning models that improve the performance/latency of the Kodiak autonomy stack which includes quantization, pruning, converting to ONNX/TensorRT, custom GPU kernels and profiling.
• Help identify and achieve parity between onboard and deployed offboard models using validation on HIL 
• Collaborate with cross-functional teams including perception, planning, simulation, and autonomy infrastructure to integrate ML models into production systems.
• Analyze real-world driving logs to identify edge cases and improve model robustness and safety.
• Use profiling tooling to additionally help improve the train time for models by identifying and removing bottlenecks.
• Contribute to the development of scalable AI infrastructure that supports continuous learning and deployment across Kodiak’s fleet.

• MS or PhD in Computer Science, Robotics, Machine Learning, or a related technical field, or equivalent practical experience.
• 6+ years experience deploying machine learning or deep learning systems in production environments.
• Proficiency in Python, C++ and modern ML frameworks such as PyTorch or TensorFlow.
• Experience working with large-scale distributed training systems.
• Solid understanding of optimization, and machine learning fundamentals.
• Strong software engineering fundamentals including testing, debugging, and system design.
• Ability to work collaboratively across teams to solve complex autonomy and robotics challenges.

What we offer:

• Competitive compensation package including equity and annual bonuses
• Excellent Medical, Dental, and Vision plans through Kaiser Permanente, Cigna, and  MetLife (including a medical plan with infertility benefits)
• MetLife Legal Services, Identity & Fraud Protection, Hospital Indemnity Insurance, Accident Insurance, & Critical Illness Insurance
• Flexible PTO, 10 paid holidays, and generous parental leave policies
• Our office is centrally located in Mountain View, CA
• Office perks: dog-friendly, free catered lunch, a fully stocked kitchen, and free EV charging
• Long Term Disability, Short Term Disability, Life Insurance
• Wellbeing Benefits - Headspace through Cigna, Calm through Kaiser, One Medical, Gympass, Spring Health through Cigna, Rula (mental health navigation) 
• Fidelity 401(k)
• Commuter, FSA, Dependent Care FSA, HSA
• Various incentive programs (referral bonuses, patent bonuses, etc.)

About this role

As a Staff Machine Learning Platform Engineer, you will help design, improve, and operate a scalable ML platform to accelerate model training, deployment, and governance. You are the technical bridge between data science and production engineering.  You’ll be joining a small but deeply critical team that scales Faire’s ability to support tens of thousands of local businesses in a constantly narrowing retail landscape.

What You Will Do

• Design and operate ML infrastructure, including workspaces, clusters, jobs, and workflows
• Productionize ML workloads using Spark, Delta Lake, MLflow, and Databricks Workflows
• Teach data scientists how to utilize our ML platform to advance development from notebook to production for our most critical models
• Implement Unity Catalog for data governance, lineage, access control, and secure multi-tenant usage
• Build CI/CD pipelines for ML using Terraform and Git-based workflows (e.g., GitHub Actions)
• Optimize performance, reliability, and cost across training and inference workloads
• Configure Identity and Access Management (IAM) and Role Based Authentication Controls (RBAC) for sensitive data sets
• Establish observability for data quality, model performance, and platform health
• Build and maintain ML Platform technical documentation

What it takes

• 8+ years of experience building production ML or data platforms
• A degree (preferably graduate level) in Computer Science, Engineering, Statistics, or a related technical field
• Strong hands-on expertise with Databricks, Spark, Delta Lake, and MLflow.
• Proficiency in Python, SQL, and distributed systems concepts
• Experience with cloud platforms and infrastructure-as-code
• Solid understanding of MLOps best practices: CI/CD, monitoring, reproducibility, and security
• Experience supporting multiple ML teams in a shared platform environment
• Are an active owner of orphaned problems and are willing to assimilate whatever knowledge you’re missing to get the job done

Tech Stack

Faire uses a modern cloud based tech stack.  For this role, you’ll want to be proficient with the following:

Salary Range

San Francisco: the pay range for this role is $246,500 to $339,000 per year. 

This role will also be eligible for equity and benefits. Actual base pay will be determined based on permissible factors such as transferable skills, work experience, market demands, and primary work location. The base pay range provided is subject to change and may be modified in the future.

The opportunity
Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company.

Our systems operate at scale across batch and streaming data, supporting analytics, product intelligence, machine learning pipelines, and business operations. As data volume and complexity grow, our platform enables large-scale model training, feature generation, and experimentation workflows that power production ML systems.

We’re looking for a Machine Learning Engineer to join our Offline Infrastructure team. This is an ideal role for a recent PhD graduate who is excited to work on large-scale systems and apply research-driven thinking to real-world machine learning problems.

You’ll help build and evolve the infrastructure that powers training data generation, ML workflows, and distributed model training. Working closely with experienced engineers and researchers, you’ll contribute to systems that ensure our ML pipelines are reliable, scalable, and efficient.

This role offers the opportunity to bridge research and production—translating advanced ideas into systems that operate at scale.

What you'll be doing

• Build and maintain data pipelines that generate training datasets for machine learning models and experimentation
• Contribute to infrastructure that supports distributed training workflows (e.g., PyTorch, Ray)
• Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to support multi-stage ML pipelines
• Improve reproducibility and reliability through dataset validation, monitoring, and testing
• Partner with ML engineers to support experimentation and model iteration
• Help optimize performance and efficiency across data processing and training systems
• Contribute to the evolution of our offline ML platform architecture as it scales

What we're looking for

• PhD in Computer Science, Machine Learning, Systems, or a related field
• Strong foundation in machine learning systems, distributed systems, or large-scale data processing (through research or projects)
• Experience with Python and working with data-intensive workloads
• Familiarity with ML frameworks (e.g., PyTorch, TensorFlow) and/or distributed systems (e.g., Ray, Spark)
• Experience (academic or applied) with data pipelines, model training workflows, or large datasets
• Strong problem-solving skills and ability to translate research ideas into practical systems
• Interest in building scalable, reliable infrastructure for machine learning
• Nice to Have
• Experience with workflow orchestration systems (Airflow, Flyte, etc.)
• Exposure to large-scale data platforms (data lakes, warehouses, streaming systems)
• Publications or research in ML systems, distributed systems, or related areas

Additional information

• Relocation support is not available for this position
• Work visa/immigration sponsorship is not available for this position

Benefits
At Unity, we want our team members to thrive. We offer a wide range of benefits designed to support well-being and work-life balance.

Please note: Benefits eligibility, specific offerings, and coverage vary based on the country and employment status.

While specific benefits vary, here are some of the ways we strive to take care of our eligible team members globally: Comprehensive health, life, and disability insurance | Commute subsidy | Employee stock ownership | Competitive retirement/pension plans | Generous vacation and personal days | Support for new parents through leave and family-care programs | Office food snacks | Mental Health and Wellbeing programs and support | Employee Resource Groups | Global Employee Assistance Program | Training and development programs | Volunteering and donation matching program

Life at Unity
Unity [NYSE: U] is the world’s leading game engine, powering play for more than 3 billion consumers each month. The top mobile games in the world, the most played PC indie titles, the most innovative console games, and virtually all of the top XR and Web Games are developed, deployed, and grown in Unity. Unity also enables teams across industries like automotive, manufacturing, and healthcare to design, simulate, and collaborate in 3D — closing the gap between ideas and reality. For more information, please visit www.unity.com.

Unity is a proud equal opportunity employer. We are committed to fostering an inclusive, innovative environment and celebrate our employees across age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, or any other protected status in accordance with applicable law. Our differences are strengths that enable us to support the growing and evolving needs of our customers, partners, and collaborators. If you have a disability that means there are preparations or accommodations we can make to help ensure you have a comfortable and positive interview experience, please fill out this form to let us know.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

This position requires the incumbent to have a sufficient knowledge of English to have professional verbal and written exchanges in this language since the performance of the duties related to this position requires frequent and regular communication with colleagues and partners located worldwide and whose common language is English.

Headhunters and recruitment agencies may not submit resumes/CVs through this Web site or directly to managers. Unity does not accept unsolicited headhunter and agency resumes. Unity will not pay fees to any third-party agency or company that does not have a signed agreement with Unity.

Your privacy is important to us. Please take a moment to review our Prospect Privacy Policy and Applicant Privacy Policy. Should you have any concerns about your privacy, please contact us at DPO@unity.com.

#MID #LI-ED1

About Scale

At Scale AI, our mission is to accelerate the development of AI applications. For 8 years, Scale has been the leading AI data foundry, helping fuel the most exciting advancements in AI, including: generative AI, defense applications, and autonomous vehicles. With our recent Series F round, we’re accelerating the abundance of frontier data to pave the road to Artificial General Intelligence (AGI), and building upon our prior model evaluation work with enterprise customers and governments, to deepen our capabilities and offerings for both public and private evaluations.

About the ACE team 

The Agent Capabilities & Environments (ACE) team, part of Scale’s Research organization, brings together customer-facing Researchers and Applied AI Engineers. Our core mission includes research on agent environments and RL reward signals, benchmarking autonomous agent performance across real-world scenarios and environments, creating robust data programs to improve Large Language Models (LLMs) agentic capabilities and building foundational tools and frameworks for evaluating models as agents. ACE focuses on autonomous agents that dynamically interact with diverse external environments, including code repositories, GUI interfaces, browsers, and more. 

About This Role

This role is at the intersection of cutting-edge AI research and practical application, with a focus on studying the data types essential for building state-of-the-art agents, such as browser and SWE agents. The ideal candidate will explore the data landscape needed to advance intelligent, adaptable AI agents, guiding the data strategy at Scale to drive innovation. This position requires not only expertise in LLM agents and planning algorithms but also creativity in addressing novel challenges related to data, interaction, and evaluation. You will contribute to impactful research publications on agents, collaborate with customer researchers, and work alongside the engineering team to translate these advancements into real-world, scalable solutions.

Ideally you’d have: 

• Practical experience working with LLMs, with proficiency in frameworks like Pytorch, Jax, or Tensorflow. You should also be adept at interpreting research literature and quickly turning new ideas into prototypes.
• A track record of published research in top ML venues (e.g., ACL, EMNLP, NAACL, NeurIPS, ICML, ICLR, COLM, etc.)
• At least three years of experience addressing sophisticated ML problems, either in a research setting or product development.
• Strong written and verbal communication skills and the ability to operate cross-functionally. 

Nice to have:

• Hands-on experience with open source LLM fine-tuning or involvement in bespoke LLM fine-tuning projects using Pytorch/Jax. 
• Hands-on experience and publications in building applications and evaluations related to AI agents such as tool-use, text2SQL, browser agents, coding agents and GUI agents.
• Hands-on experience with agent frameworks such as OpenHands, Swarm, LangGraph, etc.
• Familiarity with agentic reasoning methods such as STaR and PLANSEARCH 
• Experience working with cloud technology stack (eg. AWS or GCP) and developing machine learning models in a cloud environment.

Our research interviews are crafted to assess candidates' skills in practical ML prototyping and debugging, their grasp of research concepts, and their alignment with our organizational culture. We will not ask any LeetCode-style questions.

About Scale

At Scale AI, our mission is to accelerate the development of AI applications. For 8 years, Scale has been the leading AI data foundry, fueling the most exciting advancements in AI, including generative AI, defense applications, and autonomous vehicles. With our recent Series F round, we’re amplifying access to high-quality data to drive progress toward Artificial General Intelligence (AGI). Building on our history of model evaluation with enterprise and government customers, we are expanding our capabilities to set new standards for both public and private evaluations.

About This Role

This role operates at the forefront of AI research and real-world implementation, with a strong focus on reasoning within large language models (LLMs). The ideal candidate will study the data types critical for advancing LLM-based agents, including browser and software engineering (SWE) agents. You will play a key role in shaping Scale’s data strategy by identifying the most effective data sources and methodologies for improving LLM reasoning. Success in this role requires a deep understanding of LLMs, planning algorithms, and novel approaches to agentic reasoning, as well as creativity in tackling challenges related to data generation, model interaction, and evaluation. You will contribute to impactful research on language model reasoning, collaborate with external researchers, and work closely with engineering teams to bring state-of-the-art advancements into scalable, real-world solutions.

Ideally, you’d have:

• Practical experience working with LLMs, with proficiency in frameworks like PyTorch, JAX, or TensorFlow. You should also be skilled at rapidly interpreting research literature and turning new ideas into working prototypes.

• A track record of published research in top ML and NLP venues (e.g., ACL, EMNLP, NAACL, NeurIPS, ICML, ICLR, CoLLM, etc.).

• At least three years of experience solving complex ML challenges, either in a research setting or product development, particularly in areas related to LLM capabilities and reasoning.

• Strong written and verbal communication skills, along with the ability to work effectively across teams.

Nice to have:

• Hands-on experience fine-tuning open-source LLMs or leading bespoke LLM fine-tuning projects using PyTorch/JAX.

• Research and practical experience in building applications and evaluations related to LLM-based agents, including tool-use, text-to-SQL, browser agents, coding agents, and GUI agents.

• Experience with agent frameworks such as OpenHands, Swarm, LangGraph, or similar.

• Familiarity with advanced agentic reasoning techniques such as STaR and PLANSEARCH.

• Proficiency in cloud-based ML development, with experience in AWS or GCP environments.

Our research interviews are designed to assess candidates' ability to prototype and debug ML models, their depth of understanding in research concepts, and their alignment with our organizational culture. We do not conduct LeetCode-style problem-solving assessments.

Scale's LLM post-training platform team builds our internal distributed framework for large language model training. The platform powers MLEs, researchers, data scientists, and operators for fast and automatic training and evaluation of LLMs. It also serves as the underlying training framework for the data quality evaluation pipeline.

Scale is uniquely positioned at the heart of the field of AI as an indispensable provider of training and evaluation data and end-to-end solutions for the ML lifecycle. You will work closely with Scale’s ML teams and researchers to build the foundation platform which supports all our ML research and development works. You will be building and optimizing the platform to enable our next generation LLM training, inference and data curation.

If you are excited about shaping the future AI via fundamental innovations, we would love to hear from you!

You will:

• Build, profile and optimize our training and inference framework.
• Collaborate with ML and research teams to accelerate their research and development, and enable them to develop the next generation of models and data curation.
• Research and integrate state-of-the-art technologies to optimize our ML system.

Ideally you’d have:

• Passionate about system optimization
• Experience with multi-node LLM training and inference
• Experience with developing large-scale distributed ML systems
• Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc.
• Strong software engineering skills, proficient in frameworks and tools such as CUDA, Pytorch, transformers, flash attention, etc. 
• Strong written and verbal communication skills to operate in a cross functional team environment.

Nice to haves:

• Demonstrated expertise in post-training methods and/or next generation use cases for large language models including instruction tuning, RLHF, tool use, reasoning, agents, and multimodal, etc.

Scale’s ML platform (RLXF) team builds our internal distributed framework for large language model training and inference. The platform has been powering MLEs, researchers, data scientists and operators for fast and automatic training and evaluation of LLM's, as well as evaluation of data quality.

Scale is uniquely positioned at the heart of the field of AI as an indispensable provider of training and evaluation data and end-to-end solutions for the ML lifecycle. You will work closely across Scale’s ML teams and researchers to build the foundation platform that supports all our ML research and development. You will be building and optimizing the platform to enable our next generation of LLM training, inference and data curation.

If you are excited about shaping the future AI via fundamental innovations, we would love to hear from you!

You will:

• Build, profile and optimize our training and inference framework
• Collaborate with ML teams to accelerate their research and development and enable them to develop the next generation of models and data curation
• Research and integrate state-of-the-art technologies to optimize our ML system

Ideally you’d have:

• Strong excitement about system optimization
• Experience with multi-node LLM training and inference
• Experience with developing large-scale distributed ML systems
• Strong software engineering skills, proficient in frameworks and tools such as CUDA, Pytorch, transformers, flash attention, etc. 
• Strong written and verbal communication skills and the ability to operate in a cross functional team environment

Nice to haves:

• Demonstrated expertise in post-training methods &/or next generation use cases for large language models including instruction tuning, RLHF, tool use, reasoning, agents, and multimodal, etc.

Senior Machine Learning Engineer - Model Evaluations, Public Sector

The Public Sector ML team at Scale deploys advanced AI systems—including LLMs, agentic models, and multimodal pipelines—into mission-critical government environments. We build evaluation frameworks that ensure these models operate reliably, safely, and effectively under real-world constraints. As an ML Engineer, you will design, implement, and scale automated evaluation pipelines that help customers trust and operationalize advanced AI systems across defense, intelligence, and federal missions.

You will:

• Develop and maintain automated evaluation pipelines for ML models across functional, performance, robustness, and safety metrics, including LLM-judge–based evaluations.
• Design test datasets and benchmarks to measure generalization, bias, explainability, and failure modes.
• Build evaluation frameworks for LLM agents, including infrastructure for scenario-based and environment-based testing.
• Conduct comparative analyses of model architectures, training procedures, and evaluation outcomes.
• Implement tools for continuous monitoring, regression testing, and quality assurance for ML systems.
• Design and execute stress tests and red-teaming workflows to uncover vulnerabilities and edge cases.
• Collaborate with operations teams and subject matter experts to produce high-quality evaluation datasets.
• Comfortable with light travel (approximately 10%) for customer interaction and team needs.

This role will require an active security clearance or the ability to obtain a security clearance.

Ideally you’d have:

• Experience in computer vision, deep learning, reinforcement learning, or NLP in production settings.
• Strong programming skills in Python; experience with TensorFlow or PyTorch.
• Background in algorithms, data structures, and object-oriented programming.
• Experience with LLM pipelines, simulation environments, or automated evaluation systems.
• Ability to convert research insights into measurable evaluation criteria.

Nice to haves:

• Graduate degree in CS, ML, or AI.
• Cloud experience (AWS, GCP) and model deployment experience.
• Experience with LLM evaluation, CV robustness, or RL validation.
• Knowledge of interpretability, adversarial robustness, or AI safety frameworks.
• Familiarity with ML evaluation frameworks and agentic model design.
• Experience in regulated, classified, or mission-critical ML domains.

Scale AI is the data engine for the entire AI industry. Our mission is to accelerate the development of AI applications by providing organizations with the high-quality data they need. The Physical AI team at Scale is focused on the next frontier: building general AI that can reason and act in the physical world. By leveraging Scale’s massive data infrastructure, we are helping frontier labs build Foundation Models for Physical AI that will redefine the future of automation.

Role Overview

As the Technical Lead Manager (TLM) for the Physical AI team of Scale, you will bridge the gap between cutting-edge Machine Learning research and physical robot deployment. You will lead a high-performing team of Research Engineers while remaining a hands-on technical contributor (~60% of your time).

Your primary focus will be the development and evaluation of Large-Scale Foundation Models (e.g VLAs, World models) that allow robots and AVs to generalize across diverse tasks, environments, and morphologies.

Key Responsibilities

Technical Leadership & Research

• Model Scaling: Direct research into scaling laws for Physical AI, determining how to best utilize massive datasets for pre-training and fine-tuning generalist policies.
• VLA and World model development: Develop novel methods for developing and evaluating models, including new Physical AI industry benchmarks
• Hands-on Modeling: Actively write code to implement, train and test SOTA architectures. Conduct research on Physical AI data collection, cross-embodiment training, and policy fine-tuning.  
• Data Strategy: Collaborate with internal labeling teams to design "robotic-native" data pipelines, including the use of VLMs for automated trajectory annotation and data synthesis.
• Collaborate closely with customers to drive the industry forward in using Scale data

Team Management & Execution

• Mentorship: Lead and grow a team of 4-6 elite Physical AI  researchers, fostering a culture of high-velocity experimentation and rigorous evaluation.
• Paper-to-Product: Translate the latest research from NeurIPS, ICRA, and CVPR into production-ready features for Scale’s Physical AI partners.
• Cross-functional Alignment: Work with cross-functional teams (e.g Product and Operations) to bring our research breakthroughs into production. 

Required Qualifications

AI/ML Excellence

• Deep Learning Mastery: Expert-level proficiency in PyTorch, with deep knowledge of Transformer architectures, Attention mechanisms, and Self-Supervised Learning.
• VLM/VLA Experience: Proven track record of working with Vision-Language Models (e.g., CLIP, PaLM-E) and adapting them for spatial reasoning or embodied tasks.
• Generative AI: Experience with Diffusion Models for sequence generation or Generative World Models for predictive modeling.

Physical AI & Software Background

• Embodied AI: Strong understanding of Physical AI stack, including imitation learning, reinforcement learning (RL), and multi-modal sensor fusion.
• Infrastructure: Experience with large-scale distributed training across GPU clusters and high-performance data loading.
• Leadership: 1+ years of experience leading technical teams or projects in a research-intensive environment.

Nice to Haves:

• Publication Record: First-author publications at top-tier AI/ML conferences (NeurIPS, CVPR, ICRA, CoRL).
• Hardware Generalization: Experience building models that work across different robot types (arms, mobile bases, humanoids).
• Sim-to-Real: Experience with high-fidelity simulators (e.g., Isaac Gym, MuJoCo) and the nuances of physical domain adaptation.

About This Role

Lead applied ML engineering on Scale's Applied ML team, powering data infrastructure for leading agentic LLMs (ChatGPT, Gemini, Llama). You will build scalable multi-agent systems to validate agentic reasoning and behaviors, scale human expertise, and drive research into real-world agent reliability failures despite strong benchmarks, shipping production fixes.

Ideal for exceptional engineers with deep research rigor and a relentless focus on practical, high-impact systems. You will iterate rapidly with data, leverage AI tools to accelerate development, and collaborate tightly across engineering, product, and research.

If you excel at turning frontier agent research into reliable deployed systems, we want to hear from you.

You will:

• Build and deploy multi-agent systems for agentic reasoning validation
• Develop pipelines to detect errors and scale human judgment
• Combine classical ML, LLMs, and multi-agent techniques for reliability
• Lead research into agent failure modes and ship fixes
• Use AI tools to speed prototyping and iteration
• Build data-driven evaluations and deploy rapid improvements
• Integrate systems into Scale's platform

Ideally You’ll Have: 

• PhD or MSc in Computer Science, Mathematics, Statistics, or related field
• 3+ years shipping scaled production ML systems
• Demonstrated real-world impact
• Mastery of PyTorch, TensorFlow, JAX, or scikit-learn
• Deep expertise in agentic LLMs and multi-agent systems
• Strong software engineering and microservices (AWS/GCP)
• Rapid, data-driven iteration
• Proficiency using AI tools to accelerate work
• Strong research depth with practical bias
• Excellent cross-functional communication

Nice to Have: 

• Experience prototyping agent evaluation/reliability systems
• Human-in-the-loop or annotation pipeline work
• Open-source contributions in agents, evaluation, or alignment
• Publications on agent reliability (NeurIPS, ICML, ICLR)

The goal of a Senior Machine Learning Engineer at Scale is to leverage techniques in the fields of generative AI, computer vision, reinforcement learning, and agentic AI to improve Scale's products and customer experience in production environments. Our machine learning engineers take advantage of robust internal infrastructure and  unique access to massive datasets to deliver improvements to our customers.

Our Public Sector Machine Learning team is focused on deploying cutting-edge models to mission-critical government systems through products like Donovan and Thunderforge. Our work spans multiple modalities, with a strong focus on both large language models and computer vision. On the LLM side, we are developing agentic systems that help solve complex operational and planning challenges for government partners. This includes building agent frameworks that integrate with custom retrieval pipelines and production APIs, as well as evaluation tools to benchmark and refine agent behavior. We're also advancing research in areas like reinforcement learning for agentic LLMs, with successful deployment into real-world operational environments. On the computer vision front, we're training advanced models to increase labeling throughput and automate perception tasks. Our efforts include building large-scale fine-tuning pipelines, training models across multiple modalities, and developing generalizable vision foundation models to support a wide range of defense applications.

You will:

• Take state of the art models developed internally and from the community, use them in production to solve problems for our customers and taskers
• Improve and maintain production models through retraining, hyperparameter tuning, and architectural updates, while preserving core performance characteristics
• Collaborate with product and research teams to identify and prototype ML-driven product enhancements, including for upcoming product lines
• Work with massive datasets to develop both generic models as well as fine tune models for specific products
• Build scalable machine learning infrastructure to automate and optimize our ML services
• Serve as a cross-functional representative and advocate for machine learning techniques across engineering and product organizations
• Be comfortable learning new technologies quickly and managing multiple priorities in a fast-paced environment
• Comfortable with light travel (approximately 10%) for customer interaction and team needs
• This role will require an active security clearance or the ability to obtain a security clearance

Ideally You’d Have:

• Extensive experience with GenAI, Agentic AI, natural language processing, deep learning and deep reinforcement learning, or computer vision in a production environment
• Solid background in algorithms, data structures, and object-oriented programming
• Strong programing skills in Python, experience in Tensorflow or PyTorch

Nice to Haves:

• Graduate degree in Computer Science, Machine Learning or Artificial Intelligence specialization
• Experience working with cloud platforms (eg. AWS or GCP) and deploying machine learning models in cloud environments
• Experience with computer vision, generative AI models, large language models, or agentic systems
• Familiarity with ML evaluation frameworks and agentic model design

As a Software Engineer on the Machine Learning Infrastructure team, you will build the "Operating System" for our large-scale GPU clusters. You will architect a high-performance training platform that handles the immense complexity of multi-thousand GPU workloads, ensuring every cycle is used efficiently. Your work directly determines the velocity at which our researchers can train and iterate on the world’s most advanced models.

The ideal candidate is a systems expert who thrives on solving the orchestration, networking, and reliability challenges that emerge at massive scale. You will partner closely with researchers to build a seamless, resilient environment that transforms raw compute into breakthrough AI.

You will:

• Architect and scale a multi-tenant orchestration layer that abstracts away the complexity of GPU clusters, ensuring high utilization and seamless job recovery.
• Design and implement scheduling primitives to optimize the lifecycle of training jobs.
• Develop deep observability and automated health-checking into the training stack to proactively identify and isolate hardware failures
• Evaluate and integrate emerging technologies in the CNCF and AI ecosystem (e.g. Ray, Kueue), making data-driven build vs. buy decisions that balance velocity with long-term maintainability.
• Work closely with Finance and Procurement teams to drive our capacity planning process.
• Participate in our team’s on call process to ensure the availability of our services.
• Own projects end-to-end, from requirements, scoping, design, to implementation, in a highly collaborative and cross-functional environment.

Ideally you'd have:

• 5+ years of experience in backend or infrastructure engineering, with at least 2 years focused on orchestrating ML workloads at scale (100+ GPU nodes).
• Strong programming skills in one or more languages (e.g. Python, Go, Rust, C++)
• Experience with complex compute management systems that cover queueing, quotas, preemption, and gang scheduling.
• Experience with distributed training infrastructure, such as EFA, Infiniband, and topology-aware scheduling.
• Experience with distributed storage systems (e.g. Lustre, S3) as they relate to training throughput
• Expert-level knowledge of Kubernetes internals (Custom Resources, Operators, Admission Controllers) and how they interact with device plugins for specialized hardware.
• Familiarity with cloud infrastructure (AWS, GCP) and infrastructure as code (e.g., Terraform).
• Proven ability to solve complex problems and work independently in fast-moving environments.

Nice to haves:

• Experience with distributed training techniques such as DeepSpeed, FSDP, etc.
• Experience with the NVIDIA software and hardware stack (CUDA, NCCL)
• Experience with PyTorch
• Familiarity with post-training algorithms such as GRPO, and with Reinforcement Learning

AI is becoming vitally important in every function of our society. At Scale, our mission is to accelerate the development of AI applications. For 9 years, Scale has been the leading AI data foundry, helping fuel the most exciting advancements in AI, including generative AI, defense applications, and autonomous vehicles. With our recent investment from Meta, we are doubling down on building out state of the art post-training algorithms to reach the performance necessary for complex agents in enterprises around the world. 

The Enterprise ML Research Lab works on the front lines of this AI revolution. We are working on an arsenal of proprietary research and resources that serve all of our enterprise clients. As an ML Sys Research Engineer, you’ll work on building out the algorithms for our next-gen Agent RL training platform, support large scale training, and research and integrate state-of-the-art technologies to optimize our ML system. Your customer will be other MLREs and AAIs on the Enterprise AI team who are taking the training algorithms and applying them to client use-cases ranging from next-generation AI cybersecurity firewall LLMs to training foundation healthtech search models. If you are excited about shaping the future of the modern AI movement, we would love to hear from you!

You will: 

• Build, profile and optimize our training and inference framework.
• Post-train state of the art models, developed both internally and from the community, to define stable post-training recipes for our enterprise engagements. 
• Collaborate with ML teams to accelerate their research and development, and enable them to develop the next generation of models and data curation..
• Create a next-gen agent training algorithm for multi-agent/multi-tool rollouts.

Ideally you’d have:

• At least 1-3 years of LLM training in a production environment
• Passionate about system optimization
• Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc.
• Ability to demonstrate know-how on how to operate the architecture of the modern GPU cluster
• Experience with multi-node LLM training and inference
• Strong software engineering skills, proficient in frameworks and tools such as CUDA, Pytorch, transformers, flash attention, etc. 
• Strong written and verbal communication skills to operate in a cross functional team environment.
• PhD or Masters in Computer Science or a related field

Join our Talent Community!

We’re always looking to connect with exceptional talent. If what we’re building resonates with you, we’d love to learn more about your background and what you’re looking to do next.

If you’d like to be considered for future opportunities with us, please submit your information below. This is an expression of interest, so while we may not be in touch immediately, your information will be kept on file and reviewed as new roles open up. In the meantime, we encourage you to follow us on LinkedIn and check our careers page for updates.

We are hiring across our U.S. and London hubs, and occasionally have remote opportunities as well. If you see a role that aligns with your experience, we recommend applying directly to that posting so our team can review your application as soon as possible.

At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to cancer drug development. We do this by building and running the world’s best data and AI platform so our customers can focus on the high-value challenges that are central to their own missions.

The Databricks AI Research organization enables companies to develop AI models and agents using their own data, with technologies ranging from post-training open source LLMs to developing advanced multi-agent architectures. Databricks AI does so by producing novel science and putting it into production. Databricks AI is committed to the belief that a company’s AI models and agents are just as valuable as any other core IP, and that high-quality AI should be available to all.

Job Description 

As a Sr. Research Scientist on the Scaling team, you will be responsible for keeping up with the latest developments in deep learning and advancing the scientific frontier by creating new techniques that go beyond the state of the art. You will work together on a collaborative team of researchers and engineers with diverse backgrounds and technical training. And most importantly, you will love our customers: our goal is to make our customers successful in applying state-of-the-art LLMs and AI systems, and we encode our scientific expertise into our products to make that possible.

The Impact you will have

As a Sr. Research Scientist on the AI Research Team at Databricks, You Will: 

• Define and lead independent research agendas on foundation model efficiency in model training and reinforcement learning, conducting experiments to empirically validate hypotheses and benchmark against state-of-the-art approaches
• Drive algorithmic innovations for large-scale neural network training or inference (e.g., novel optimizers, low-precision techniques, model adaptation methods)
• Optimize ML systems for distributed training, memory efficiency, and compute efficiency through hands-on implementation.

What We Look for

• MS/PhD in Computer Science or related field with strong foundations in machine learning and systems
• Proven ability to write high-quality, efficient code in Python and PyTorch for research implementation and experimentation
• Strong preference for candidates with first-author publications at top ML/systems conferences (ICLR, ICML, NeurIPS, MLSys) focused on optimization or efficiency.

About the Role

Together AI is seeking a Machine Learning Engineer to join our Inference Engine team, focusing on optimizing and enhancing the performance of our AI inference systems. This role involves working with state-of-the-art large language models models and ensuring they run efficiently and effectively at scale. If you are passionate about AI inference, PyTorch, and developing high-performance systems, we want to hear from you. This position offers the chance to collaborate closely with AI researchers and engineers to create cutting-edge AI solutions. Join us in shaping the future at Together AI!

Responsibilities

• Design and build the production systems that power the Together AI inference engine, enabling reliability and performance at scale.
• Develop and optimize runtime inference services for large-scale AI applications.
• Collaborate with researchers, engineers, product managers, and designers to bring new features and research capabilities to the world.
• Conduct design and code reviews to ensure high standards of quality.
• Create services, tools, and developer documentation to support the inference engine.
• Implement robust and fault-tolerant systems for data ingestion and processing.

Requirements

• 3+ years of experience writing high-performance, well-tested, production-quality code.
• Proficiency with Python and PyTorch.
• Demonstrated experience in building high performance libraries and tooling.
• Excellent understanding of low-level operating systems concepts including multi-threading, memory management, networking, storage, performance, and scale.
• Preferred: Knowledge of existing AI inference systems such as TGI, vLLM, TensorRT-LLM, Optimum
• Preferred: Knowledge of AI inference techniques such as speculative decoding.
• Preferred: Knowledge of CUDA/Triton programming.
• Nice to have: Knowledge of Rust, Cython and compilers.

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society. Together, we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI. Our team has been behind technological advancements such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers and engineers in our journey to build the next-generation AI infrastructure.

Compensation

We offer competitive compensation, startup equity, health insurance, and other competitive benefits. The US base salary range for this full-time position is $160,000 - $230,000 + equity + benefits. Our salary ranges are determined by location, level, and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunities to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

About the Role

At Together.ai, we are building state-of-the-art infrastructure to enable efficient and scalable inference for large language models (LLMs). Our mission is to optimize inference frameworks, algorithms, and infrastructure, pushing the boundaries of performance, scalability, and cost-efficiency.

We are seeking an Inference Frameworks and Optimization Engineer to design, develop, and optimize distributed inference engines that support multimodal and language models at scale. This role will focus on low-latency, high-throughput inference, GPU/accelerator optimizations, and software-hardware co-design, ensuring efficient large-scale deployment of LLMs and vision models.

This role offers a unique opportunity to shape the future of LLM inference infrastructure, ensuring scalable, high-performance AI deployment across a diverse range of applications. If you're passionate about pushing the boundaries of AI inference, we’d love to hear from you!

Responsibilities

Inference Framework Development and Optimization

• Design and develop fault-tolerant, high-concurrency distributed inference engine for text, image, and multimodal generation models.
• Implement and optimize distributed inference strategies, including Mixture of Experts (MoE) parallelism, tensor parallelism, pipeline parallelism for high-performance serving.
• Apply CUDA graph optimizations, TensorRT/TRT-LLM graph optimizations, and PyTorch-based compilation (torch.compile), and speculative decoding to enhance efficiency and scalability.

Software-Hardware Co-Design and AI Infrastructure

• Collaborate with hardware teams on performance bottleneck analysis, co-optimize inference performance for GPUs, TPUs, or custom accelerators.
• Work closely with AI researchers and infrastructure engineers to develop efficient model execution plans and optimize E2E model serving pipelines.

Requirements

Must-Have:

• Experience:
  • 3+ years of experience in deep learning inference frameworks, distributed systems, or high-performance computing.
• Technical Skills:
  • Familiar with at least one LLM inference frameworks (e.g., TensorRT-LLM, vLLM, SGLang, TGI(Text Generation Inference)).
  • Background knowledge and experience in at least one of the following: GPU programming (CUDA/Triton/TensorRT), compiler, model quantization, and GPU cluster scheduling.
  • Deep understanding of KV cache systems like Mooncake, PagedAttention, or custom in-house variants.
• Programming:
  • Proficient in Python and C++/CUDA for high-performance deep learning inference.
• Optimization Techniques:
  • Deep understanding of Transformer architectures and LLM/VLM/Diffusion model optimization.
  • Knowledge of inference optimization, such as workload scheduling, CUDA graph, compiled, efficient kernels
• Soft Skills:
  • Strong analytical problem-solving skills with a performance-driven mindset.
  • Excellent collaboration and communication skills across teams.

Nice-to-Have:

• Experience in developing software systems for large-scale data center networks with RDMA/RoCE
• Familiar with distributed filesystem(e.g., 3FS, HDFS, Ceph)
• Familiar with open source distributed scheduling/orchestration frameworks, such as Kubernetes (K8S)
• Contributions to open-source deep learning inference projects.

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers in our journey in building the next generation AI infrastructure.

Compensation

We offer competitive compensation, startup equity, health insurance and other competitive benefits. The US base salary range for this full-time position is: $160,000 - $230,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

Please see our privacy policy at https://www.together.ai/privacy  

About the Role

Our team focuses on enabling custom models and dedicated inference on Together. We are responsible for building a container platform, optimizing autoscaling, minimizing cold starts, achieving the best end-to-end model performance, and providing a best-in-class developer experience with great tooling. We often focus on video or audio generation across the stack: CUDA kernels, pytorch optimization, inference engines, container orchestration, queueing theory, etc. An ideal candidate will be great at profiling/optimization but know the word kubernetes, or be intimately familiar with multi-cluster scheduling and have some sense of ML bottlenecks.

Responsibilities

• New hires may work on multi-cluster orchestration, portfolio optimization, predictive autoscaling, control panes, model bring-up, model optimization, APIs for managing deployments, inference worker SDKs, and CLI tools.
• Analyze and improve the robustness and scalability of existing distributed systems, APIs, databases, and infrastructure
• Partner with product teams to understand functional requirements and deliver solutions that meet business needs
• Write clear, well-tested, and maintainable software and IaC for both new and existing systems
• Conduct design and code reviews, create developer documentation, and develop testing strategies for robustness and fault tolerance

Requirements

• 5+ years of demonstrated experience in building large scale, fault tolerant, distributed systems.
• Experience running serverless inference platforms, doing model bring-up on short notice, being on call, or running a cloud provider is a very big plus
• Good taste and ability to thoughtfully discuss how what you’ve built has failed over time
• Experience designing, analyzing and improving efficiency, scalability, and stability of various system resources
• Excellent understanding of low level operating systems concepts including concurrency, networking and storage, performance and scale
• Expert-level programmer in one or more of Python, Golang, Rust, C++, or Haskell
• Proficiency in writing and maintaining Infrastructure as Code (IaC) using tools like Terraform
• Experience with Kubernetes internals or other container orchestration systems
• Sound judgement for when to use and when to not use LLMs for code
• Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or a related technical field, or equivalent practical experience
• Writing-heavy roles or companies are a plus

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers and engineers in our journey in building the next generation AI infrastructure.

Compensation

We offer competitive compensation, startup equity, health insurance and other competitive benefits. The US base salary range for this full-time position is: $160,000 - $250,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

Please see our privacy policy at https://www.together.ai/privacy  

About the Role

Together AI is building the best inference infrastructure for voice applications. Our Voice AI platform powers production-grade, real-time voice agents and applications — serving speech-to-text and text-to-speech models with best-in-class latency and reliability.

We're looking for a Senior ML Engineer to drive the model serving layer for voice workloads. You'll work hands-on with inference engines like TRT-LLM and SGLang to optimize how we serve models like Whisper, Parakeet, Orpheus, and Kokoro — pushing latency and throughput to the frontier. You'll profile GPU utilization, design batching strategies for streaming audio, and ensure new model architectures can go from research to production quickly.

This is a foundational hire on a small, high-impact team. Voice inference has unique challenges — streaming audio, tokenization, real-time latency budgets — that require dedicated ML engineering focus. You'll shape how Together serves voice models as the industry moves from pipeline architectures (ASR → LLM → TTS) toward end-to-end speech-to-speech.

• Own the model serving stack that powers Together's voice platform across STT, TTS, and speech-to-speech.
• Work directly with state-of-the-art accelerators (H100s, H200s, B200s) to optimize voice model inference.
• Collaborate with model partners (Cartesia, Deepgram, Rime, and others) to bring their models to production on Together's infrastructure.
• Build quality evaluation frameworks that guide model selection for customers and inform the roadmap.
• Join a small, early-stage team with outsized impact on a fast-growing product area.

Responsibilities

• Optimize inference performance for voice models (STT, TTS, speech-to-speech) — targeting best-in-class TTFB, throughput, and GPU utilization across our curated model set.
• Productionize voice models on serverless and dedicated endpoints, including batching strategies, streaming inference, and memory management tailored to audio workloads.
• Build and maintain a voice model evaluation framework — measuring WER across accents, languages, and noise conditions for STT; naturalness, latency, and pronunciation accuracy for TTS.
• Enable new model architectures in our serving stack as the field evolves, including audio-native LLMs, codec-based models (SNAC), and speech-to-speech systems.
• Collaborate with model partners to integrate and optimize their models (Cartesia, Deepgram, Rime, and others) running on Together's infrastructure.
• Profile and debug performance across the full inference stack — from GPU kernels to framework-level bottlenecks — and ship measurable improvements.
• Work with the platform engineering side of the team to ensure the serving layer meets the latency and reliability requirements of real-time voice APIs.
• Contribute to voice model fine-tuning capabilities (STT and TTS) as we enable customers to build differentiated voice experiences on Together.
• Lay the groundwork for multiple new products down the line.

Requirements

• 5+ years of experience in ML engineering, with a focus on model serving, inference optimization, or ML infrastructure.
• Hands-on experience with LLM serving engines (vLLM, SGLang, TensorRT-LLM, or similar) — comfortable reading and modifying engine internals, not just using APIs.
• Strong proficiency in Python and PyTorch; experience with GPU profiling and optimization (CUDA, memory management, kernel-level debugging).
• Track record of shipping ML systems to production with measurable performance improvements.
• Strong product sense — you think about what developers building voice apps actually need, not just what's technically interesting.
• Comfort working on a small, early-stage team where you'll wear multiple hats and move fast.
• Experience with speech and audio ML (ASR, TTS architectures, audio signal processing) is a strong plus but not required — you can learn this quickly if you have strong ML engineering fundamentals.
• Familiarity with audio codecs and tokenization schemes (SNAC, Encodec, DAC) is a plus.
• Experience training or fine-tuning speech models is a plus.
• Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related field, or equivalent practical experience

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers and engineers in our journey in building the next generation AI infrastructure.

Compensation

We offer competitive compensation, startup equity, health insurance and other competitive benefits. The US base salary range for this full-time position is: $200,000 - $260,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

Please see our privacy policy at https://www.together.ai/privacy  

About the Role

Together AI is building the Inference Platform that powers the world's most advanced generative AI models. Your role will be a critical bridge between cutting-edge research and real-world applications, focusing on making translating our internal model training research to production-ready deployment for our customers. This involves a deep commitment to data-centric development, meticulous hyperparameter tuning, and rigorous checkpoint evaluation before models ever hit production.

This role will involve understanding customer specific needs and fine-tuning models on our internal data recipe and their proprietary data. The goal is to transform general-purpose models into highly performant, specialized tools that solve real business problems. You will not be training foundation models from scratch but rather focusing on creating highly efficient, specialized models by working with dedicated GPU clusters.

Responsibilities

• Design and iterate on novel speculator algorithms, combining architectural innovations with carefully curated data to push the frontier of accuracy–efficiency tradeoffs.
• Be the critical link between raw data and a production-ready model, seeing your work directly impact our customers' success.
• Work in a fast-paced, high-impact role at the cutting edge of generative AI.
• Collaborate with a team of experts dedicated to solving real-world, high-performance challenges.
• You'll collaborate directly with customers to understand their needs, and work closely with our core inference and Applied ML research teams to integrate your work into the production platform.
• A culture of deep technical ownership where you are empowered to take on and solve challenging problems

Requirements

• A genuine love for data curation and processing, with a meticulous attention to detail. You believe that great models start with great data.
• Demonstrated ability to perform effective hyperparameter searches and understand the trade-offs involved in tuning models for specific tasks.
• Experience working with and building on top of existing training codebases. You are comfortable navigating complex code and contributing to its improvement.
• Strong attention-to-detail in evaluating model checkpoints to ensure they meet strict quality, performance, and reliability standards.
• Experience with Python and PyTorch.
• Familiarity with SLURM and/or Kubernetes clusters and experience submitting and managing jobs in a high-performance computing environment.
• Familiarity with modern LLMs and generative models.
• Basic understanding of distributed training frameworks (e.g., FSDP, DeepSpeed).
• Bachelor’s, Master’s degree, or Ph.D. in Computer Science, Computer Engineering, or a related field, or equivalent practical experience.

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, ATLAS, and RedPajama. We invite you to join a passionate group of researchers and engineers in our journey in building the next generation AI infrastructure.

Compensation

We offer competitive compensation, startup equity, health insurance and other competitive benefits. The US base salary range for this full-time position is: $190,000 - $270,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

Please see our privacy policy at https://www.together.ai/privacy

What We're Looking For

As we continue to expand our footprint, we are looking for a sharp, agile, and business-minded lawyer to join our lean legal team. You will be the primary engine behind our commercial contracting process. This is not a "siloed" role; you will be the cross-functional glue between Finance, Engineering, Sales, and Security, ensuring our legal strategy moves at the speed of our innovation. 

This is an ideal opportunity for a high-performing lawyer with a strong grasp of transactional fundamentals who is ready to "level up." While your primary focus will be owning the commercial contract lifecycle, you will have a front-row seat to the broader legal challenges of a hyper-growth AI company.

You will be joining the Legal Team and report to our VP of Legal. This is a hybrid role based in our San Francisco or New York office with an in-office requirement of 2 days per week.

What You’ll Do

• Lead the drafting and negotiation of a wide variety of contracts, including Master Service Agreements (MSAs), NDAs, Data Processing Addenda (DPAs), and vendor agreements.
• Serve as the legal point of contact for CX, Sales, Engineering, and Finance teams to streamline deal cycles and mitigate risk.
• Navigate complex technology transactions, specifically focusing on intellectual property rights and data privacy frameworks in the context of AI and cloud computing.
• Beyond commercial matters, you will assist the VP of Legal on special projects, potentially spanning corporate governance, employment, and regulatory compliance as the company scales.

What You’ll Need

• 3–5 years of legal practice, with a focus on commercial or corporate transactions.
• A proven track record at a top-tier law firm, in-house at a high-growth tech company, or a strategic mix of both.
• A deep understanding of technology transactions, including the nuances of IP licensing and data privacy provisions.
• You are a confident negotiator who knows when to hold the line and when to find a creative solution to get the deal done.

Compensation

What We’re Looking For

We are looking for a Senior Finance Associate to drive priorities across strategic finance, corporate development, and FP&A as Lightning AI scales. You will partner with cross-functional stakeholders to shape unit economics, advise on deal pricing, and guide investment decisions. You will support GPU financing, fundraising, and investor relations to help Lightning AI acquire and allocate capital effectively. You will also build an operating cadence for P&L management, reporting, forecasting, and planning, turning data into decision-ready insights for leadership. 

The finance scope is deep, leadership visibility is high, and the work directly influences revenue growth, margin expansion, and capital allocation. If you want meaningful ownership, steep learning, and proximity to the decisions that shape an AI cloud, this role is for you.

This is a hybrid role based in our San Francisco or New York office with an in-office requirement of 2 days per week.

What You’ll Do

• Support deal reviews with Sales and leadership and recommend pricing decisions
• Partner with Marketing on spend allocation, pipeline efficiency, and unit economics analyses (LTV/CAC, payback period)
• Partner with Product on ROI for software investments and strategic initiatives
• Partner with Infrastructure Operations on capacity planning and CapEx assessment
• Model GPU financing structures and assess returns
• Support lender and investor communications, materials, and data rooms
• Monitor industry trends and the competitive landscape to identify potential business risks and opportunities
• Contribute to M&A sourcing and diligence models
• Build and maintain a driver based model for P&L, balance sheet, and cash flow statement
• Prepare executive-level business reviews for leadership and board
• Lead monthly and quarterly reporting, ensuring accuracy and highlighting key business drivers
• Develop forecast cycles and strengthen forecast accuracy by partnering closely with cross-functional stakeholders
• Support annual planning and long-range planning 
• Conduct rigorous analysis of key performance indicators, operational metrics, and business trends
• Manage FP&A systems and tooling and maintain data accuracy

What You’ll Need

• Bachelor's degree in Business Administration, Finance, Economics, or a related field
• 4+ years in Strategic Finance, FP&A, investment banking, private equity, or venture capital, ideally with a focus on software, AI, and/or cloud infrastructure
• Advanced proficiency in Excel and Google Sheets and experience building complex financial models
• Excellent written and verbal communication
• Track record of cross-functional partnership
• Analytical and strong problem-solving ability
• Team player with a growth mindset, positive attitude, and strong work ethic
• Bias to action and desire to work in a startup environment
• Experience with capital markets, M&A, or CapEx financing
• Exposure to cloud or AI infrastructure economics and data center capacity planning
• Familiarity with FP&A systems and data tooling (e.g., NetSuite)

Compensation

What We're Looking For

As a Machine Learning Solutions Engineer, you will operate at the intersection of machine learning, distributed systems, and cloud infrastructure.
You will partner with customers to design and deploy end-to-end AI systems, spanning:

• Model development and training
• GPU infrastructure and cluster design
• Distributed inference and production deployment

This role goes beyond traditional ML solutions engineering—you will act as a technical architect, helping customers make critical decisions across compute, orchestration, and system design.

The role can be based out of our New York City or San Francisco office, with an in-office requirement of at least 2-3 days per week and occasional team and company offsites. We are not able to provide visa sponsorship for this role at this time. The annual base pay range for this role is $150,000 - $195,000, in addition to a variable pay component and meaningful equity. 

What You’ll Do

Customer Architecture & Technical Leadership

• Partner with customers to understand ML workloads, infrastructure constraints, and scaling requirements
  
  
• Architect end-to-end solutions across:
  
  
• Data pipelines (CPU → GPU workflows)
  
  
• Distributed training (multi-node, multi-GPU)
  
  
• High-throughput inference systems
  
  
• Translate business goals (latency, cost, throughput) into technical system design decisions
  
  

GPU & Infrastructure Design

• Design and optimize workloads across GPU clusters (H100, H200, B200, etc.)
  
  
• Advise on:
  
  
• Training vs inference cluster design
  
  
• Interconnect choices (Ethernet vs Infiniband / RDMA vs Roce)
  
  
• Storage strategies (local NVMe vs networked / object storage)
  
  
• Model and optimize for:
  
  
• Tokens/sec, tokens/$
  
  
• Throughput vs latency tradeoffs
  
  
• GPU utilization and scheduling efficiency
  
  

Kubernetes & Platform Systems

• Design and support deployments on Kubernetes (EKS, GKE, on-prem clusters)
  
  
• Work with:
  
  
• GPU scheduling (time-slicing, MIG, bin-packing)
  
  
• Autoscaling and workload orchestration
  
  
• Helm-based deployments and multi-tenant environments
  
  
• Help customers balance:
  
  
• Raw Kubernetes flexibility vs platform abstraction (Lightning)
  
  

Demos, POCs, and Execution

• Build and deliver technical demos and POCs that showcase:
  
  
• Distributed training workflows
  
  
• Scalable inference endpoints
  
  
• End-to-end ML pipelines on Lightning AI
  
  
• Scope and lead POCs aligned to customer success metrics (latency, cost, reliability)
  
  

Cross-Functional Impact

• Act as the bridge between customers, product, and engineering
  
  
• Provide feedback on:
  
  
• Platform gaps in infrastructure, orchestration, and performance
  
  
• Emerging patterns in GPU usage and distributed systems
  
  
• Influence roadmap across ML workflows and infrastructure capabilities
  
  

Enablement & Thought Leadership

• Create technical content
• Architecture guides (e.g., high-throughput LLM inference systems)
• Best practices for GPU utilization and scaling
• Educate customers on modern AI infrastructure patterns

What You’ll Need

ML + Systems Expertise

• 3–6+ years experience in:
  
  
• Machine Learning / AI Engineering
  
  
• Solutions Engineering / Sales Engineering / ML Consulting
  
  
• Strong understanding of:
  
  
• Training vs inference workloads
  
  
• Model optimization (quantization, batching, caching, etc.)
  
  

GPU & Distributed Systems

• Experience working with:
  
  
• GPU clusters (NVIDIA stack preferred)
  
  
• Distributed training or inference systems
  
  
• Familiarity with:
  
  
• NCCL, CUDA, or GPU performance profiling
  
  
• Networking concepts (RDMA, Roce, Infiniband, high-throughput systems)
  
  

Kubernetes & Cloud Platforms

• Hands-on experience with:
  
  
• Kubernetes (EKS, GKE, or on-prem)
• Slurm 
• Containerization (Docker)
  
  
• Exposure to:
  
  
• GPU scheduling in Kubernetes environments
  
  
• Multi-tenant or production ML deployments
  
  

Programming & Tooling

• Strong Python skills (PyTorch preferred)
  
  
• Experience building:
  
  
• ML pipelines
  
  
• APIs or inference services
  
  
• Familiarity with Lightning AI, PyTorch Lightning, or similar frameworks is a plus
  
  

Customer-Facing Excellence

• • Ability to:
    
    
    • Explain complex infrastructure and ML tradeoffs clearly
      
      
    • Run technical discovery and uncover quantifiable success metrics
      
      
  • Experience working cross-functionally with:
    
    
    • Sales, product, and engineering teams

What We're Looking For

Lightning AI is seeking a Senior Storage Infrastructure Engineer to join our Infrastructure Engineering team.

In this role, you will focus on building and operating the storage systems that power large-scale AI/ML training, inference, and HPC workloads. You will work at the intersection of software, hardware, and operations—developing automation, improving reliability, and scaling distributed storage systems across our bare-metal infrastructure.

You will help own the data plane of our storage infrastructure, supporting high-throughput, low-latency data access for some of the most demanding AI workloads. You’ll play a key role in managing and evolving our storage stack (including VAST and S3-compatible systems like Ceph), ensuring performance, reliability, and efficiency at scale.

We’re flexible on location for this team. This role can work hybrid out of one of our US-based hubs (Seattle, NYC, or SF) or fully remote within the U.S., with occasional company and team offsites. We are not able to provide visa sponsorship for this position at this time.

What You'll Do

Storage Systems & Infrastructure

• Operate and scale distributed storage systems, including VAST and S3-compatible object storage (e.g., Ceph)
• Improve performance, reliability, and efficiency of storage systems supporting large-scale AI/ML workloads
• Troubleshoot complex storage and data path issues across hardware and software layers
• Optimize storage performance to support high-throughput, low-latency AI training and inference workloads

Automation & Tooling

• Build and maintain automation for provisioning, managing, and monitoring storage infrastructure
• Develop Python-based tools and workflows to reduce manual operational overhead
• Improve lifecycle management of storage clusters, from deployment through maintenance and scaling

Systems & Operations

• Manage and operate Linux-based systems in production, including bare-metal environments
• Partner with infrastructure and data center teams on hardware bring-up, upgrades, and issue resolution
• Support capacity planning, utilization tracking, and forecasting for storage systems
• Leverage monitoring and telemetry to diagnose issues and improve system performance and reliability

Cross-Functional Collaboration

• Work closely with Infrastructure Engineering, Network Engineering, and Platform teams to integrate storage into the broader platform
• Contribute to design discussions around new infrastructure deployments and scaling strategies
• Help define best practices for operating storage systems in high-performance computing environments

What You'll Need

Required Qualifications

• 5+ years of experience in infrastructure engineering, systems engineering, or related roles
• Hands-on experience operating distributed storage systems (e.g., VAST, Ceph, or similar)
• Strong Linux systems experience in production environments
• Proficiency in Python or similar scripting/programming languages for automation
• Experience working with bare-metal infrastructure and hardware-oriented systems
• Ability to debug complex issues across system boundaries (storage, OS, hardware, networking)
• Experience with storage networking protocols (e.g., NFS or similar)
• Experience with capacity planning, monitoring, and performance tuning

Ideal Experience

• Experience with VAST storage systems in production environments
• Experience operating S3-compatible object storage at scale
• Data center operations experience, including working with physical hardware
• Familiarity with AI/ML or HPC workloads and their storage requirements
• Background in high-performance or low-latency distributed systems
• Familiarity with high-performance data transfer technologies (e.g., RDMA, GPU Direct Storage)
• Experience supporting GPU-based workloads or large-scale compute clusters

Compensation

What We’re Looking For

Lightning AI is seeking an Observability Infrastructure Engineer to join our Infrastructure Engineering team.

In this role, you will own and evolve observability systems across large-scale, GPU-enabled bare-metal infrastructure. You’ll operate at the intersection of infrastructure, data, and product, building platforms for metrics, logs, traces, and alerting that power both internal operations and customer-facing visibility.

You will play a key role in productizing observability, enabling scalable, multi-tenant monitoring experiences while keeping pace with rapid infrastructure buildouts. This includes designing telemetry pipelines, improving signal quality, and delivering actionable insights that ensure reliability and transparency across our platform.

We’re flexible on location for this team. This role can work hybrid out of one of our US-based hubs (Seattle, NYC, or SF) or fully remote within the U.S., with occasional company and team offsites. We are not able to provide visa sponsorship for this position at this time.

What You’ll Do

Observability Platform & Productization

• Own and evolve a scalable observability platform spanning metrics, logs, traces, and events
• Drive the productization of observability capabilities for both internal teams and external customers
• Design multi-tenant observability systems with scoped access, RBAC, and customer-facing visibility
• Continuously improve observability systems to keep pace with rapid infrastructure buildouts

Telemetry & Data Pipelines

• Design and operate telemetry pipelines ingesting data from GPUs, CPUs, networking (Ethernet & InfiniBand), containers, APIs, and BMC/Redfish
• Build systems to correlate signals across infrastructure layers to enable faster debugging and root cause analysis
• Implement streaming and real-time data pipelines using tools such as Kafka, OTEL, Promtail, or similar

Alerting, Reliability & Insights

• Design and implement noise-resistant alerting systems to improve signal quality and reduce operational load
• Create dashboards and alerting for InfraOps, Engineering, and Customer Success teams
• Build automated insights and enable proactive detection, forecasting, and system health visibility at scale

Systems & Infrastructure Engineering

• Contribute to broader infrastructure engineering projects beyond observability
• Partner with infrastructure and platform teams to embed observability into core systems and workflows
• Support large-scale, distributed systems across compute, networking, and storage environments

Cross-Functional Collaboration

• Work closely with customer-facing teams to deliver external observability experiences
• Collaborate with engineering, operations, and support teams to improve system transparency and reliability
• Help define best practices for observability across the organization

What You’ll Need

Required Qualifications

• 5+ years of experience in infrastructure engineering, SRE, or observability-focused roles
• Strong experience with monitoring systems such as Prometheus, Grafana, ELK, or VictoriaMetrics
• Experience building and operating observability platforms at scale
• Proficiency in Python, Go, or bash for automation and data integration
• Familiarity with containerized environments and Kubernetes observability
• Experience with streaming telemetry pipelines (Kafka, OTEL, Promtail, or equivalent)
• Experience with multi-tenant monitoring architectures
• Strong written and verbal communication skills

Ideal Experience

• Experience with GPU observability, particularly NVIDIA DCGM
• Experience monitoring large-scale GPU or HPC clusters
• Familiarity with InfiniBand fabric observability
• Experience building customer-facing or productized infrastructure systems
• Experience with correlation engines, RCA workflows, or predictive alerting systems
• Broad exposure to infrastructure domains including networking, storage, and provisioning

Compensation

What We're Looking For

Lightning AI is seeking a GPU & Compute Infrastructure Engineer to join our Infrastructure Engineering team.

In this role, you will own image management, system diagnostics, and validation across large-scale bare-metal compute infrastructure, with a particular focus on GPU-enabled systems. You will work at the intersection of hardware, systems, and software—developing automation, improving reliability, and enabling efficient cluster bring-up for AI/ML and HPC workloads.

You will play a key role in owning and evolving our image pipeline, running validation environments and test clusters, and supporting both system-level and GPU hardware qualification. This role is critical to ensuring that our infrastructure is consistent, performant, and ready to support demanding AI workloads from day one.

We’re flexible on location for this team. This role can work hybrid out of one of our US-based hubs (Seattle, NYC, or SF) or fully remote within the U.S., with occasional company and team offsites. We are not able to provide visa sponsorship for this position at this time.

What You'll Do

Systems, Image & Validation Infrastructure

• Own and evolve systems for image management, deployment, and validation across bare-metal infrastructure
• Run and maintain test clusters used for system validation, diagnostics, and bring-up
• Validate firmware, drivers, and OS images across compute and GPU-enabled systems
• Support hardware qualification efforts for next-generation platforms

GPU Diagnostics & Performance

• Own GPU diagnostics and validation workflows across large-scale infrastructure
• Diagnose and resolve complex issues across GPUs, drivers, OS, and hardware layers
• Analyze system and GPU performance using tools such as NVIDIA DCGM
• Identify failure patterns and drive improvements in system stability and validation coverage

Automation & Tooling

• Build and maintain automation for provisioning, validation, and system bring-up
• Develop Python-based tools and workflows to improve efficiency and reduce manual operational overhead
• Improve the reliability, repeatability, and scalability of image pipelines and validation systems

Systems & Operations

• Manage and operate Linux-based systems in production and validation environments
• Manage virtualization technology
• Support bare-metal provisioning workflows, including PXE and image-based systems
• Interface with hardware management systems (e.g., IPMI, Redfish) for monitoring and debugging

Cross-Functional Collaboration

• Partner with Infrastructure, Hardware, and Data Center teams on system bring-up and validation
• Collaborate with platform and ML teams to ensure systems meet workload requirements
• Contribute to best practices for provisioning, diagnostics, and lifecycle management of infrastructure

What You'll Need

Required Qualifications

• 5+ years of experience in infrastructure engineering, systems engineering, or related roles
• Strong Linux systems experience in production environments
• Hands-on experience with GPU-enabled systems and tools such as NVIDIA DCGM
• Familiarity with bare-metal provisioning and system bring-up workflows
• Proficiency in Python or similar scripting/programming languages for automation
• Ability to debug complex issues across hardware, OS, GPUs, and system software

Ideal Experience

• Experience with high-performance interconnects (e.g., InfiniBand, NVLink)
• Experience with PXE boot environments, LiveCD systems, or image-based provisioning workflows
• Experience with hardware management interfaces such as iDRAC, IPMI, or Redfish
• Data center operations experience, including working with physical hardware
• Experience supporting AI/ML or HPC workloads at scale
• Experience with GPU validation frameworks or large-scale hardware qualification processes

Compensation

What We're Looking For

We’re looking for a Fullstack Engineer to help develop and scale the Lightning AI platform across the frontend, CLI, APIs, and core backend systems. In this role, you’ll take ownership of key features and drive them end-to-end, working closely with engineering, product, and design to ship high-quality, iterative improvements. This is an opportunity to build systems that support real-world machine learning workloads at scale.

This role is based in one of our hubs (NYC, SF, Seattle, or London), with a minimum of 2 in-office days per week and occasional team and company offsites. We are not able to provide visa sponsorship for this position at this time.

What You’ll Do

• Write readable/testable/efficient code in React and either Python or Go (Golang).
• Build deep expertise in our technology stack to deliver new features, improve system stability, and increase overall performance.
• Partner with engineering and product leaders on developing the Lightning product while using your experience to set the technical direction for large projects.
• Evaluate, strengthen, and document technical architecture, tools, and processes.
• Maintain high standards for software quality, implement automation, drive continuous delivery, and reduce time to production while proactively reducing technical debt.
• Mentor and coach engineers on system design, operating in high uncertainty, and problem-solving to create a supportive, inclusive environment where each engineer can grow.

What You’ll Need

• Proficiency in React and either Go (Golang) or Python, with a preference for Go
• Strong understanding of software engineering principles and lifecycle
• Hands-on experience as a Fullstack Engineer in a SaaS technology company
• Proven ability to take ownership of key features and drive end-to-end development
• Communication and collaboration with engineers, product managers, and designers
• Ability to operate in high uncertainty and rapidly changing environments

Compensation

What We're Looking For

We’re looking for a Frontend Engineer to help develop and scale the UI and frontend infrastructure of the Lightning AI platform. In this role, you’ll take ownership of key features and drive them end-to-end, working closely with engineering, product, and design to ship high-quality, iterative improvements. With over 10,000 organizations building with Lightning, you’ll have a unique opportunity to impact how the world builds and deploys AI to production.

This role is based in one of our hubs (Seattle, NYC, SF, or London), with a minimum of 2 in-office days per week and occasional team and company offsites. We are not able to provide visa sponsorship for this position at this time.

What You’ll Do

• Write readable/testable/efficient code in React and Redux, and build deep expertise in our technology stack to deliver new features, improve system stability, and increase overall performance.
• Partner with engineering and product leaders on developing the user interface and frontend infrastructure while using your experience to set the technical direction for large projects.
• Evaluate, strengthen, and document technical architecture, tools, and processes.
• Maintain high standards for software quality, implement automation, drive continuous delivery, and reduce time to production while proactively reducing technical debt.
• Mentor and coach engineers on system design, operating in high uncertainty, and problem-solving to create a supportive, inclusive environment in which each engineer can grow.

What You’ll Need

• Proficiency in React, JavaScript and TypeScript.
• Strong understanding of software engineering principles and lifecycle.
• Hands-on experience as a Frontend Engineer in a SaaS technology company.
• Proven ability to take ownership of key features and drive end-to-end development.
• Communication and collaboration with engineers, product managers, and designers.
• Ability to operate in high uncertainty and rapidly changing environments.

Compensation

What We're Looking For

We’re looking for a Backend Engineer to help develop and scale the Lightning AI platform, spanning APIs, infrastructure, billing, security, and integrations. In this role, you’ll take ownership of key features and drive them end-to-end, working closely with engineering, product, and design to ship high-quality, iterative improvements. This is an opportunity to build systems that support real-world machine learning workloads at scale.

This role is based in one of our hubs (Seattle, NYC, SF, or London), with a minimum of 2 in-office days per week and occasional team and company offsites. We are not able to provide visa sponsorship for this position at this time.

What You’ll Do

• Write readable/testable/efficient code in Go (Golang).
• Build deep expertise in our technology stack to deliver new features, improve system stability, and increase overall performance.
• Partner with engineering and product leaders on developing the Lightning product while using your experience to set the technical direction for large projects.
• Evaluate, strengthen, and document technical architecture, tools, and processes.
• Maintain software quality, implement automation, drive continuous delivery, and reduce time to production while proactively reducing technical debt.
• Mentor and coach engineers on system design, operating in high uncertainty, and problem-solving to create a supportive, inclusive environment where each engineer can grow.

What You’ll Need

• Extensive experience writing efficient, maintainable code in Go. Familiarity with best practices in testing, readability, and optimization is essential.
• Broad knowledge of backend and frontend systems, with experience building and scaling APIs, CLI tools, billing systems, and security protocols. Exposure to frontend frameworks is a plus.
• Experience with cloud platforms like AWS, GCP, or Azure and designing systems that are reliable, performant, and scalable. Familiarity with containerization (Docker, Kubernetes) is a strong advantage.
• Proven ability to take ownership of key features and drive end-to-end development
• Commitment to software quality, test automation, and continuous delivery. Experience proactively managing technical debt and speeding up time to production.
• Hands-on experience as a backend or product engineer in a SaaS technology company
• Communication and collaboration with engineers, product managers, and designers
• Ability to operate in high uncertainty and rapidly changing environments

Compensation

We are looking for a Senior Software Engineer specializing in Machine Learning to join our Revenue ML team at Discord. This team partners with our revenue product groups, focusing on both consumer revenue and our emerging Ads initiative. This role will specifically contribute to our Ads ML efforts, helping to build and scale ML capabilities in areas such as ads measurement, targeting, and delivery ranking.

As part of this team, you will play a critical role in developing foundational ML models that enhance ad relevance, optimize performance, and drive revenue. This is a unique opportunity to work on an early-stage Ads ML platform and have a direct impact on the business's success. Our tech stack includes Python, ML frameworks like PyTorch and TensorFlow, large-scale data infrastructure, and real-time ad-serving technologies.

What You'll Be Doing:

• Design, develop, and deploy machine learning models for ads targeting and ranking.
• Develop sophisticated ML solutions such as identity graph to enhance ad targeting.
• Build and optimize ad ranking models to serve the most effective ads based on campaign objectives (e.g., app installs, link click).
• Improve ads targeting and ranking by leveraging both on-platform and off-platform signals.
• Collaborate cross-functionally with product, engineering, and business teams to define and execute on the Ads ML roadmap.
• Scale our ML infrastructure to support an increasing number of concurrent ad campaigns while ensuring low-latency decision-making.
• Drive research and implementation of state-of-the-art ML techniques in the field of online advertising.

What You Should Have:

• 5+ years of experience as a Machine Learning Engineer or Data Scientist.
• 3+ years of experience specifically in Ads ML (ads ranking, personalization, optimization, privacy-compliant user modeling, targeting, or measurement).
• Strong proficiency in Python and familiarity with deep learning frameworks such as PyTorch or TensorFlow.
• Experience with applied deep learning (e.g transformers, embedding models).
• Proven track record of designing, implementing, and scaling ML-driven ad systems in real-world applications.
• Experience working with real-time ML inference, A/B testing, and optimization frameworks.
• Experience translating ML evaluation results and performance metrics into actionable product roadmap items.
• Ability to connect business objectives to ML solutions, with the flexibility to shift focus toward the highest-impact problems as priorities evolve.

Bonus Skills:

• Strong understanding of performance advertising and how ML impacts revenue and advertiser retention.
• Knowledge of ad tech industry standards and ads ecosystem including targeting, retrieval, ranking, pacing, frequency, auction, etc.
• Experience with large-scale recommendation systems.
• Experience with large-scale data infrastructure and distributed computing

The US base salary range for this full-time position is $220,000 to $247,000 + equity + benefits. Our salary ranges are determined by role and level. Within the range, individual pay is determined by additional factors, including job-related skills, experience, and relevant education or training. Please note that the compensation details listed in US role postings reflect the base salary only, and do not include equity, or benefits.

What we offer:

• Competitive compensation package including equity and annual bonuses
• Excellent Medical, Dental, and Vision plans through Kaiser Permanente, Cigna, and  MetLife (including a medical plan with infertility benefits)
• MetLife Legal Services, Identity & Fraud Protection, Hospital Indemnity Insurance, Accident Insurance, & Critical Illness Insurance
• Flexible PTO, 10 paid holidays, and generous parental leave policies
• Our office is centrally located in Mountain View, CA
• Office perks: dog-friendly, free catered lunch, a fully stocked kitchen, and free EV charging
• Long Term Disability, Short Term Disability, Life Insurance
• Wellbeing Benefits - Headspace through Cigna, Calm through Kaiser, One Medical, Gympass, Spring Health through Cigna, Rula (mental health navigation) 
• Fidelity 401(k)
• Commuter, FSA, Dependent Care FSA, HSA
• Various incentive programs (referral bonuses, patent bonuses, etc.)

What We're Looking For

We are looking for an Account Executive with experience selling highly technical, developer-focused products to CTOs, VPs of Engineering, software engineers, researchers, and/or data scientists. You thrive in a fast-paced, start-up environment focusing on every step of the sales process, from prospecting, qualifying, and nurturing leads to running demos, negotiating contracts, and closing deals. We want someone who thinks about how to uncover and address our customers’ problems to drive customer acquisition while feeding their requirements back to inform the product roadmap.

This role is based out of our New York City office, with an in-office requirement of at least 3 days per week and occasional team and company offsites.

What You’ll Do

• Build an understanding of our target markets, value proposition, and pricing models to not only meet but exceed quarterly revenue targets.
• Focus on high-quality, consistent outbound prospecting across key market segments to build the sales pipeline and optimize the customer funnel.
• Qualify and develop both inbound and outbound sales leads.
• Collaborate with the product, engineering, and marketing teams to continuously improve and iterate on the sales process.
• Analyze industry trends and customer segmentation to identify and capitalize on new sources of qualified leads.
• Work alongside the marketing team to increase the number and quality of inbound leads.
• Champion data integrity and accuracy using HubSpot to track activity and manage sales opportunities through the funnel.

What You’ll need

• 5+ years of experience as an Account Executive or in a similar sales role, selling technical or developer-focused products to technical audiences.
• Familiarity with the machine learning, AI, or software development ecosystem, and an ability to grasp complex technical concepts.
• Proven track record of meeting or exceeding revenue targets in a fast-paced startup environment.
• Experience with outbound prospecting, qualifying leads, running demos, and closing deals is essential.
• Ability to uncover customer pain points, articulate value propositions, and effectively address objections to drive solutions.
• Strong interpersonal skills to work cross-functionally with leadership, product, engineering, and marketing teams.
• Proficiency with HubSpot or a similar CRM to track activity, manage the sales pipeline, and ensure data integrity.
• Solid understanding and application of a sales methodology (e.g., MEDDIC, Challenger, or Sandler) with the ability to tailor approaches to different customer needs and scenarios.

Compensation

This role offers $300K OTE, with a 50/50 base/variable split, in addition to meaningful equity. 

The opportunity
Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company.

Our systems operate at scale across batch and streaming data, supporting analytics, product intelligence, machine learning pipelines, and business operations. As data volume and complexity grow, our platform also supports large-scale model training, feature generation, and experimentation workflows that power production ML systems.

To support this growth, we need strong technical ownership to ensure our ML pipelines remain reliable, scalable, and architecturally sound.

We are seeking a staff ML engineer to design and evolve the large-scale offline platform. This role focuses on building reliable infrastructure for generating training datasets, orchestrating ML workflows, and enabling efficient, distributed model training at scale. You will work closely with ML engineers and platform teams to ensure our pipelines can efficiently handle growing data volumes and increasingly complex training workloads.

You will play a key role in shaping how model datasets are prepared as well as model training, validated, and delivered to distributed training systems, while ensuring the reliability, scalability, and performance of our offline ML platform.

What you'll be doing

• Design and operate large-scale data pipelines that generate training datasets used for machine learning training and experimentation
• Develop infrastructure that supports distributed training workflows using technologies such as Pytorch, Ray Data, and Ray Train, etc.
• Integrate ML pipelines with workflow orchestration systems (e.g., Flyte, Airflow, or similar) to enable reliable multi-stage training workflows
• Improve reproducibility and observability of ML pipelines through dataset validation, monitoring, and automated testing
• Optimize performance and resource utilization across distributed compute systems used for data processing and model training
• Partner closely with ML engineers to enable efficient large-scale experimentation and model iteration
• Lead architectural improvements to ensure our offline ML pipelines remain scalable, reliable, and cost-efficient

What we're looking for

• Strong experience building large-scale ML pipelines
• Experience working with distributed computing frameworks such as Ray, Spark, Flink and familiarity in the Ray ecosystem (Ray Data, Ray Train) for distributed data processing and model training
• Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines
• Deep experience designing and operating production-grade data pipelines
• Strong programming skills in Python and experience working with large-scale distributed workloads
• Experience with modern data infrastructure (data lakes, warehouses, orchestration systems, streaming platforms)
• Strong systems thinking, with the ability to reason about performance, scalability, reliability, and cost tradeoffs in distributed systems
• Proven ability to lead technical direction and influence architectural decisions across teams without formal authority

Additional information

• Relocation support is not available for this position

Benefits
At Unity, we want our team members to thrive. We offer a wide range of benefits designed to support well-being and work-life balance.

Please note: Benefits eligibility, specific offerings, and coverage vary based on the country and employment status.

While specific benefits vary, here are some of the ways we strive to take care of our eligible team members globally: Comprehensive health, life, and disability insurance | Commute subsidy | Employee stock ownership | Competitive retirement/pension plans | Generous vacation and personal days | Support for new parents through leave and family-care programs | Office food snacks | Mental Health and Wellbeing programs and support | Employee Resource Groups | Global Employee Assistance Program | Training and development programs | Volunteering and donation matching program

Life at Unity
Unity [NYSE: U] is the world’s leading game engine, powering play for more than 3 billion consumers each month. The top mobile games in the world, the most played PC indie titles, the most innovative console games, and virtually all of the top XR and Web Games are developed, deployed, and grown in Unity. Unity also enables teams across industries like automotive, manufacturing, and healthcare to design, simulate, and collaborate in 3D — closing the gap between ideas and reality. For more information, please visit www.unity.com.

Unity is a proud equal opportunity employer. We are committed to fostering an inclusive, innovative environment and celebrate our employees across age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, or any other protected status in accordance with applicable law. Our differences are strengths that enable us to support the growing and evolving needs of our customers, partners, and collaborators. If you have a disability that means there are preparations or accommodations we can make to help ensure you have a comfortable and positive interview experience, please fill out this form to let us know.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

This position requires the incumbent to have a sufficient knowledge of English to have professional verbal and written exchanges in this language since the performance of the duties related to this position requires frequent and regular communication with colleagues and partners located worldwide and whose common language is English.

Headhunters and recruitment agencies may not submit resumes/CVs through this Web site or directly to managers. Unity does not accept unsolicited headhunter and agency resumes. Unity will not pay fees to any third-party agency or company that does not have a signed agreement with Unity.

Your privacy is important to us. Please take a moment to review our Prospect Privacy Policy and Applicant Privacy Policy. Should you have any concerns about your privacy, please contact us at DPO@unity.com.

#SEN #LI-ED1

About The Role

The AI team is building cutting-edge solutions that bring the power of AI directly to edge devices while seamlessly integrating with cloud infrastructure. We are looking for a Software Engineer to design and develop high-performance, scalable services to support AI workloads across edge and cloud environments.

What You Might Do

• Contribute to design, build, and maintain services that power AI-driven applications, ensuring scalability and performance.
• Develop APIs and microservices that facilitate seamless integration between cloud-based AI models and edge devices.
• Optimize data pipelines and storage solutions for real-time AI inference and processing.
• Work closely with AI researchers, infrastructure engineers, and frontend developers to deliver end-to-end AI-driven solutions.
• Build and optimize an agent orchestration runtime that enables tool use, memory management, and multi-step reasoning across LLMs, APIs, and edge-connected systems.
• Support the implementation of security and privacy best practices for distributed AI systems.

Essential Qualifications

• Requires at least Masters in Computer Science, Engineering, Data Science, Artificial Intelligence or related fields.
• 1+ years of experience in backend software development, with strong proficiency in Python, Java, and C++. 
• Working knowledge in LLM integration into multi-agent systems with strong understanding of agent orchestration, tool use, and memory/context management.
• Comfortable tinkering with LLMs via prompting and experienced with tools use, task decomposition, and structured outputs.
• Familiarity with database management systems (SQL and NoSQL) and data streaming technologies.
• Strong problem-solving skills and ability to work in a fast-paced, agile environment.

Preferred Skills

• Experience with AI model deployment and integration into backend services.
• Exposure to cloud-edge hybrid architectures and distributed AI workloads.
• Familiarity with on-device AI frameworks such as PyTorch, TensorFlow Lite, ONNX Runtime, or CoreML.

Salary Range: $127,000 - $175,000

About the Role
We are seeking a Biomedical AI Engineer to design and deploy agentic AI systems that can reason over biomedical knowledge, plan multi-step analyses, and autonomously integrate diverse scientific data sources to accelerate therapeutic discovery.

This role goes beyond traditional LLM + RAG pipelines. You will build AI agents that can retrieve, synthesize, evaluate, and act on biomedical information across literature, omics datasets, and clinical records operating as intelligent systems embedded within discovery workflows.

Key Responsibilities

• Design and implement agentic AI architectures capable of multi-step reasoning, planning, and tool use in biomedical contexts.
• Develop LLM-based agents that dynamically retrieve, rank, and synthesize knowledge from scientific literature, knowledge graphs, omics datasets, and clinical data.
• Build memory and feedback mechanisms that allow agents to refine hypotheses, update context, and adapt to evolving data.
• Integrate external tools (analysis pipelines, simulation engines, statistical models, databases) into autonomous agent workflows.
• Architect scalable systems that combine structured and unstructured biomedical data within production-grade AI environments.
• Collaborate closely with computational biologists, translational scientists, and software engineers to embed agentic systems into real-world discovery pipelines.
• Evaluate robustness, reliability, and interpretability of autonomous systems in high-stakes biomedical settings.

Qualifications

• Degree in Computer Science, Machine Learning, Computational Biology, Biomedical Informatics, or related field.
• Strong hands-on experience with large language models, including fine-tuning, alignment, and structured prompting.
• Experience building agent-based systems or complex LLM orchestration frameworks (e.g., multi-agent systems, tool-using LLMs, planning modules).
• Proficiency in Python and modern ML frameworks (PyTorch, JAX, TensorFlow).
• Experience working with large-scale biomedical datasets (genomics, transcriptomics, clinical records, or scientific corpora).

Preferred

• Experience designing autonomous research assistants or AI systems that perform multi-step scientific reasoning.
• Familiarity with knowledge graphs, hybrid symbolic-neural systems, or multimodal foundation models.
• Understanding of privacy, security, and regulatory considerations in healthcare AI.

Compensation

We offer a competitive compensation and benefits package, seeking to provide an open, flexible, and friendly work environment to empower employees and provide them with a platform to develop their long-term careers. A Summary of Benefits is available for all applicants. We offer a competitive package that includes base salary, bonus, and equity. The base pay range for this position is expected to be $190,000 - $280,000 annually; however, the base pay offered may vary depending on the market, job-related knowledge, skills and capabilities, and experience.

Xaira Therapeutics an equal-opportunity employer. We believe that our strength is in our differences. Our goal to build a diverse and inclusive team began on day one, and it will never end.

TO ALL RECRUITMENT AGENCIES: Xaira Therapeutics does not accept agency resumes. Please do not forward resumes to our jobs alias or employees. Xaira Therapeutics is not responsible for any fees related to unsolicited resumes.

• Develop multimodal foundation models spanning image and molecular data
• Design approaches that unify representation learning and generative modeling
• Train large-scale models on structured, high-dimensional datasets
• Explore cross-modal alignment, prediction, and generation
• Build scalable systems for data processing, training, and evaluation
• Work in a research-driven environment with evolving problem definitions

• Experience with representation learning (self-supervised, contrastive, masked modeling, etc.)
• Experience with generative modeling (diffusion, autoregressive models, VAEs, or related approaches)
• Proficiency in Python, ML frameworks, and CI/CD (PyTorch, JAX, etc.)

• Work on multimodal learning or cross-domain representation learning
• Experience with image data, molecular / omics-like data, or both
• Familiarity with large-scale training

• Exposure to biological or scientific datasets
• Experience with paired or partially aligned multimodal data
• Comfortable working in open-ended, evolving problem spaces

Compensation

We offer a competitive compensation and benefits package, seeking to provide an open, flexible, and friendly work environment to empower employees and provide them with a platform to develop their long-term careers. A Summary of Benefits is available for all applicants. We offer a competitive package that includes base salary, bonus, and equity. The base pay range for this position is expected to be $170,000 - $240,000 annually; however, the base pay offered may vary depending on the market, job-related knowledge, skills and capabilities, and experience.

Xaira Therapeutics an equal-opportunity employer. We believe that our strength is in our differences. Our goal to build a diverse and inclusive team began on day one, and it will never end.

TO ALL RECRUITMENT AGENCIES: Xaira Therapeutics does not accept agency resumes. Please do not forward resumes to our jobs alias or employees. Xaira Therapeutics is not responsible for any fees related to unsolicited resumes.

About the Role
We are looking for an AI Scientist to join our team at the intersection of machine learning and single-cell biology. This role will focus on developing and applying foundation models to large-scale perturb-seq and single-cell datasets to uncover new biological mechanisms and therapeutic opportunities. You will help design scalable ML architectures that can extract insight from high-dimensional perturbation data and integrate across multi-omic modalities.

Key Responsibilities

• Design, train, and refine foundation models tailored to perturb-seq and other single-cell datasets.
  Build pipelines for preprocessing, normalization, and integration of large-scale single-cell and perturbation data.
• Apply ML/AI approaches to model cellular responses to perturbations and identify causal biological insights.
• Collaborate with experimental biologists to align computational methods with experimental design and discovery goals.
• Develop multimodal integration approaches across transcriptomic, epigenomic, and proteomic single-cell readouts.
• Publish methods and results in top scientific journals and present at conferences.
  
  

Qualifications

• PhD or equivalent in Computational Biology, Computer Science, Bioinformatics, or related discipline.
• Strong expertise in single-cell data analysis, particularly perturb-seq 
• Hands-on experience with foundation models or large-scale self-supervised learning (e.g., transformers, variational autoencoders).
• Strong coding skills in Python, ML frameworks (PyTorch/TensorFlow), and single-cell analysis tools (Scanpy, Seurat).
• Experience in developing scalable ML methods for large biological datasets.
  
  

Preferred

• Background in causal inference, generative models, or representation learning for biological systems.
• Familiarity with multi-omics integration and cross-modal foundation models.
• Track record of impactful publications in computational biology, machine learning, or related fields.

Compensation

We offer a competitive compensation and benefits package, seeking to provide an open, flexible, and friendly work environment to empower employees and provide them with a platform to develop their long-term careers. A Summary of Benefits is available for all applicants. We offer a competitive package that includes base salary, bonus, and equity. The base pay range for this position is expected to be $150,000 - $240,000 annually; however, the base pay offered may vary depending on the market, job-related knowledge, skills and capabilities, and experience.

Xaira Therapeutics an equal-opportunity employer. We believe that our strength is in our differences. Our goal to build a diverse and inclusive team began on day one, and it will never end.

TO ALL RECRUITMENT AGENCIES: Xaira Therapeutics does not accept agency resumes. Please do not forward resumes to our jobs alias or employees. Xaira Therapeutics is not responsible for any fees related to unsolicited resumes.

Location: San Francisco, CA (Hybrid) 

What is Verse? 

Energy markets are more volatile than ever. Rapid electrification and the rise of AI are driving unprecedented demand for power, while energy costs continue to rise across the globe. For the world’s largest energy buyers, managing energy has never been more complex or more critical.

Verse helps these organizations manage complex power portfolios with confidence by unifying energy data, planning, forecasting, and operations in one tool. Our Energy Cost Intelligence platform, Aria, brings together energy, finance, and operations teams with real-time, finance-ready intelligence—replacing spreadsheets and consultants with precision across the entire energy lifecycle. Built by an expert team of energy buyers, data scientists, and engineers, Verse enables faster, smarter energy decisions that reduce risk and lower energy costs.

The Role

Verse is seeking a Data Scientist to join our Data Science Team. In this role, you will lead the development and deployment of advanced data-driven solutions across a range of applications, including electricity markets, renewable procurement, and energy risk management. You will shape the machine learning and data modeling foundations that Verse's software is built on. For example, you might spend a cycle deploying electricity market price forecasting pipelines for new regions, developing solar production anomaly detection models, or creating scalable tools for benchmarking energy project financial performance.

Key Responsibilities

• Lead End-to-End Data Science Projects: Own and drive projects from problem definition through scoping, modeling, validation, and production deployment. Translate business problems into scalable, high-impact modeling solutions.
• Statistical & Machine Learning Modeling: Design, develop, and refine statistical and machine learning models (e.g., time series forecasting, probabilistic models) to support decision-making and enhance product capabilities.
• Analytics Engineering & Data Modeling: Perform complex data transformations and develop well-structured data models. Translate business and analytical requirements into scalable, tested, and well-documented datasets.
• Software Development & Productionization: Write clean, efficient, and maintainable Python code. Contribute to integrating models into production systems in a cloud-based environment while leveraging AI coding tools to accelerate development.
• MLOps: Contribute to Verse’s machine learning modeling infrastructure to support scaling of ML models and improving reliability, monitoring, and performance in production.
• Cross-Functional Collaboration: Partner with product, engineering, and business stakeholders to ensure models and insights are aligned with user needs and effectively integrated into workflows.

What We're Looking For (Minimum Qualifications)

• Master’s degree in Computer Science, Statistics, Engineering, Applied Mathematics, or a related quantitative field and 2+ years of professional experience in data science or machine learning; or
• Bachelor’s degree and 4+ years of professional experience in data science or machine learning
• Strong foundation in statistical modeling and machine learning, including time series forecasting and model evaluation
• Experience deploying and maintaining models in cloud-based environments (e.g., AWS, GCP, or Azure) using MLOps practices
• Strong Python expertise, including experience with scientific computing and ML libraries (e.g., NumPy, pandas, scikit-learn, PyTorch, TensorFlow)
• Hands-on experience in orchestrating complex data transformations (e.g. Airflow, Dagster, dbt)
• Strong software engineering practices (version control, testing, code reviews, CI/CD)
• Strong communication skills, with the ability to explain technical concepts to non-technical stakeholders

What Will Make You Standout (Preferred Qualifications)

• Experience in energy, climate tech, or related domains (not required)
• Familiarity with optimization methods or operations research
• Experience developing probabilistic forecasting models and quantifying uncertainty
• PhD in a quantitative field

What Makes Verse a Great Place to Work? 

• Lead with Empathy: We lift each other up with humility and kindness, always putting colleagues and customers first
• Be Honest & Transparent: We prioritize effective communication to build trust with our team, customers, and stakeholders
• Move with Balance & Precision: We believe speed and perseverance must be accompanied by thoughtfulness and reflection
• Leave the World a Better Place: We are passionate about our mission, and we strive to create a sustainable world for future generations

Base Pay Range

$146,000 - $172,000

This is the estimated base salary range for this position, which does not include the value of benefits or a potential equity grant. A wide range of factors are considered in making compensation decisions, including but not limited to skill sets, market conditions, experience and training, licensure and certifications, and business and organizational needs.

Benefits and Employee Perks 

• Competitive compensation and equity grant at a high-growth start up 
• Comprehensive benefits package including medical, dental and vision insurance, and 401k 
• Flexible hours and unlimited PTO 
• Diverse and inclusive working environment 

Verse is an equal opportunity employer. All applicants and employees are considered for hire, promotion, and compensation without regard to race, color, religion, sex, national origin, age, disability, sexual orientation, marital or familial status.

About the Role

Our Engineering team creates innovative product experiences that power personalized marketing at massive scale. We develop cutting-edge applications and systems that process 100s of billions of messages and events per year, enabling marketers to connect effectively with hundreds of millions of consumers. Joining our team offers a high-growth career opportunity to work with some of the world’s most talented engineers in a high-performance and high-impact culture.

As a Senior Software Engineer in our Intelligent Messaging, you will pioneer the future of personalized 1:1 marketing by architecting robust, scalable backend systems powered by advanced AI and LLMs, including cutting-edge Generative AI and GPT technologies. Collaborating with top-tier engineers, data scientists, product managers, and designers, you’ll craft the foundation and reshape the platform of multi-channel messaging experimentation to support innovative, AI-driven applications that empower marketers to forge deep, meaningful connections with consumers at scale. Your impactful work will drive the delivery of billions of tailored messages, transforming consumer shopping experiences and fueling success for the world’s leading brands.

What You’ll Accomplish

• Develop and maintain scalable backend systems for our customer-facing products, ensuring high performance and reliability.
• Collaborate with cross-functional teams to build applications that support multi-channel communications for marketers.
• Implement and optimize systems that process and deliver billions of messages daily.
• Improve code quality through code reviews, testing, and advocating for best practices.
• Identify and address technical debt to ensure the long-term health of our codebase.
• Drive architectural design and technical decisions, and stay current with emerging technologies to enhance our products.

Your Expertise

• 5+ years of professional experience in software development focusing on backend systems.
• Proficiency in Java, Python, or Go, with a strong understanding of object-oriented programming.
• Experience building and maintaining scalable, high-performance applications.
• Proven ability to collaborate effectively with cross-functional teams.
• Solid understanding of software development best practices, including code reviews, writing tests, and continuous integration.

Nice to Haves

• Experience with service-oriented architecture and distributed systems.
• Familiarity with AWS services and cloud infrastructure.
• Knowledge of databases such as DynamoDB, Postgres, or Redis.
• Experience with messaging systems or streaming platforms (e.g., Kafka, Pulsar).
• Familiarity with frontend development with React and TypeScript.
• Experience with DevOps practices and tools such as Docker and Kubernetes.

What We Use

• Our infrastructure runs primarily in Kubernetes hosted in AWS’s EKS.
• Infrastructure tooling includes Istio, Datadog,Terraform, CloudFlare, and Helm.
• Our backend is Java / Spring Boot microservices, built with Gradle, coupled with things like DynamoDB, Kinesis, AirFlow, Postgres, Planetscale, and Redis, hosted via AWS.
• Our frontend is built with React and TypeScript, and uses best practices like GraphQL, Storybook, Radix UI, Vite, esbuild, and PlaywrightOur automation is driven by custom and open source machine learning models, lots of data and built with Python, Metaflow, HuggingFace 🤗, PyTorch, TensorFlow, and Pandas.

You'll get competitive perks and benefits, from health & wellness to equity, to help you bring your best self to work.

For US based applicants:

• The US base salary range for this full-time position is $190,000 - $230,000 annually + equity + benefits
• Our salary ranges are determined by role, level and location

#LI-EZ1

By applying for this position, your data will be processed as per Attentive's Privacy Policy.

At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to cancer drug development. We do this by building and running the world’s best data and AI platform so our customers can focus on the high-value challenges that are central to their own missions.

The Mosaic AI organization enables companies to develop AI models and systems using their own data, with technologies ranging from pre-training LLMs from scratch to augmented generation using the latest retrieval techniques. Mosaic AI does so by producing novel science and putting it into production. Mosaic AI is committed to the belief that a company’s AI models are just as valuable as any other core IP, and that high-quality AI models should be available to all.

Job Description

As a research engineer on the Scaling team, you will be responsible for keeping up with the latest developments in deep learning and advancing the scientific frontier by creating new techniques that go beyond the state of the art. You will work together on a collaborative team of researchers and engineers with diverse backgrounds and technical training. And most importantly, you will love our customers: our goal is to make our customers successful in applying state-of-the-art LLMs and AI systems, and we encode our scientific expertise into our products to make that possible.

The Impact you will have

As a research engineer on the Scaling Team at Databricks, you will: 

• Drive performance improvements through advanced optimization techniques including kernel fusion, mixed precision, memory layout optimization, tiling strategies, and tensorization for training-specific patterns
• Design, implement, and optimize high-performance GPU kernels for training workloads (e.g., attention mechanisms, custom layers, gradient computation, activation functions) targeting NVIDIA architectures
• Design and implement distributed training frameworks for large language models, including parallelism strategies (data, tensor, pipeline, ZeRO-based) and optimized communication patterns for gradient synchronization and collective operations
• Profile, debug, and optimize end-to-end training workflows to identify and resolve performance bottlenecks, applying memory optimization techniques like activation checkpointing, gradient sharding, and mixed precision training.

What We Look for

• BS/MS/PhD in Computer Science or related field with hands-on experience writing and tuning CUDA kernels for ML training applications, or hands-on experience in distributed training frameworks (PyTorch DDP, DeepSpeed, Megatron-LM, FSDP) 
• Strong understanding of NVIDIA GPU architecture (memory hierarchy, tensor cores, warp scheduling, SM occupancy) and proficiency with CUDA debugging/profiling tools (Nsight, NVProf)
• Deep understanding of parallelism techniques and memory optimization strategies for large-scale model training, with proven ability to debug and optimize distributed workloads
• Strong software engineering skills in Python and PyTorch, with experience supporting production training workflows and knowledge of LLM training dynamics including hyperparameter tuning and optimization strategies.

We are looking for a Senior Machine Learning Engineer to join our Revenue ML team at Discord. This role sits at the intersection of Discord's two most strategic revenue pillars — our growing 1P Shop and our newly launched Game Commerce platform, which connects players to in-game items from major publishers like Marvel Rivals, Fortnite, Valorant, and more. You'll be the founding ML voice for commerce discovery and personalization, building systems from the ground up that power recommendations, social commerce mechanics, and marketing targeting across both first-party and third-party storefronts.

This is a high-impact, high-leverage role. Discord's social platform gives us a fundamentally differentiated commerce advantage — deep social graphs, fan communities, and native gaming context — and you will be the person who turns that into ML-powered products that drive meaningful GMV growth.

What You'll Be Doing:

• Architect and own the ML foundations for commerce discovery: user, item, and interaction embeddings that power personalized recommendations across shop surfaces (homepage, cart, post-purchase, wishlist, and more).
• Design and deploy scalable real-time recommendation and ranking systems that support a growing catalog of 1P and 3P items across heterogeneous game publisher inventories.
• Build ML-powered marketing targeting systems that identify the right users for the right campaigns — new buyer discounts, drop campaigns, weekly deals, and seasonal promotions — driving conversion without conditioning users to wait for discounts.
• Leverage Discord's unique social graph to build social commerce ML: gifting recipient prediction, group buying conversion modeling, and friend-group recommendations that differentiate Discord from traditional game storefronts.
• Drive deep learning A/B testing infrastructure and model monitoring to translate experimentation results into actionable product decisions.
• Partner closely with Shop, Game Commerce, Revenue Infra, ML Infra and Data Engineering teams to define ML requirements, surface integration points, and influence the commerce roadmap.

What You Should Have:

• 4+ years of experience as a Machine Learning Engineer, with a track record of owning and shipping recommendation or personalization systems end-to-end.
• Deep expertise in applied deep learning — particularly embedding models, two-tower architectures, and retrieval/ranking systems for e-commerce or content recommendation.
• Strong proficiency in Python and deep learning frameworks (PyTorch preferred).
• Experience building and operating real-time ML serving infrastructure at scale, including feature stores, model serving, and A/B testing frameworks.
• Demonstrated ability to work in early-stage, high-ambiguity environments and build ML systems from the ground up, not just improve existing ones.
• Experience translating ML evaluation metrics and experiment results into product roadmap decisions and business impact.
• Strong cross-functional instincts — you're comfortable partnering with product, engineering, data science, and business stakeholders to align on priorities and drive execution.

Bonus Skills:

• Experience applying graph ML or social network signals (social affinities, community behavior) to recommendation or personalization problems.
• Familiarity with personalized marketing systems: lifecycle targeting, audience segmentation, and campaign optimization.
• Familiarity with loyalty, rewards, or incentive pro

The US base salary range for this full-time position is $220,000 to $247,500 + equity + benefits. Our salary ranges are determined by role and level. Within the range, individual pay is determined by additional factors, including job-related skills, experience, and relevant education or training. Please note that the compensation details listed in US role postings reflect the base salary only, and do not include equity, or benefits.

Who We Are:
The Machine Learning Platform team at Reddit is a high-impact team that owns the infrastructure that powers recommendations, content discovery, user and content quantification, while directly impacting other teams such as Growth, Ads, Feeds, and Core Machine Learning teams.

What You’ll Do:
As a Staff Machine Learning Engineer, you will lead the development of a large-scale ML Inference Platform at Reddit.

• Lead the end-to-end design, implementation, and maintenance of a highly available, low-latency GPU-based model serving system for search, ranking, and LLMs supporting Millions of QPS.
• Design and develop ML and Generative AI systems in cloud-based production environments on Kubernetes at scale.
• Rapidly develop prototypes and develop a high-performance feature hydration and processing system as a part of the inference stack - including routing, caching, and batching.
• Lead a unified GPU model export framework to support converting trained models into optimized GPU inference models.
• Strong understanding of real-time ML observability to track feature/model performance.
• Experience working with LLM serving online at scale.
• Built an E2E inference performance benchmarking framework
• Deep Understanding of multi-cluster compute environment and network topology that is specific to ML inference use cases.

Who You Might Be:

• 7+ years of experience in ML Engineering, AI Platform Engineering, or Cloud AI Deployment roles.
• Have experience operating orchestration systems such as Kubernetes at scale
• Deep experience with cloud-based technologies for supporting an ML platform, including tools like AWS, Google Cloud Storage, infrastructure-as-code (Terraform), and more
• Proficiency with the common programming languages and frameworks of ML, such as Go, Python, etc.
• Excellent communication skills with the ability to articulate technical AI concepts to non-technical stakeholders
• Strong focus on scalability, reliability, performance, and ease of use. You are an undying advocate for platform users and have a deep intuition for the genAI product development lifecycle.
• Strong knowledge of model serving, inference pipelines, monitoring, and observability for AI systems is a plus
• Strong proficiency in Python and deep experience with modern AI/ML frameworks (Triton, Dynamo, vLLM, Pytorch)

Benefits:

• Comprehensive Healthcare Benefits and Income Replacement Programs
• 401k with Employer Match
• Global Benefit programs that fit your lifestyle, from workspace to professional development to caregiving support
• Family Planning Support
• Gender-Affirming Care
• Mental Health & Coaching Benefits
• Flexible Vacation & Paid Volunteer Time Off
• Generous Paid Parental Leave 

• Design and develop multimodal transformer architectures that fuse camera, LiDAR, and radar into unified representations
• Research and implement cross-modal attention mechanisms, token fusion strategies, and efficient multi-stream tokenization
• Build scalable training pipelines for large-scale multimodal transformers across massive real-world datasets
• Explore self-supervised and contrastive pretraining objectives that learn transferable multimodal representations
• Optimize transformer models for real-time inference under latency and compute constraints

• BS, MS, or PhD in AI, Computer Science, or a related field
• 4+ years experience with transformer architectures, particularly in multimodal or multi-stream settings
• Familiarity with cross-attention, token fusion, or modality alignment techniques
• Proficiency in Python and deep learning frameworks like PyTorch or TensorFlow
• Strong understanding of scalable training for large models, including distributed training and mixed-precision optimization
• Passion for building AI that reasons over the full breadth of sensory input to operate safely in the real world

What we offer:

• Competitive compensation package including equity and annual bonuses
• Excellent Medical, Dental, and Vision plans through Kaiser Permanente, Cigna, and  MetLife (including a medical plan with infertility benefits)
• MetLife Legal Services, Identity & Fraud Protection, Hospital Indemnity Insurance, Accident Insurance, & Critical Illness Insurance
• Flexible PTO, 10 paid holidays, and generous parental leave policies
• Our office is centrally located in Mountain View, CA
• Office perks: dog-friendly, free catered lunch, a fully stocked kitchen, and free EV charging
• Long Term Disability, Short Term Disability, Life Insurance
• Wellbeing Benefits - Headspace through Cigna, Calm through Kaiser, One Medical, Gympass, Spring Health through Cigna, Rula (mental health navigation) 
• Fidelity 401(k)
• Commuter, FSA, Dependent Care FSA, HSA
• Various incentive programs (referral bonuses, patent bonuses, etc.)

What You’ll Do:

The Account Management team at Weights & Biases (W&B) by CoreWeave, drives retention and revenue growth across our existing customer base by ensuring AI/ML teams achieve meaningful outcomes with our platform. We partner closely with Field Engineering—who leads technical success—to connect product adoption to business value, reduce friction in ML workflows, and uncover whitespace across organizations. Our team builds deep, multi-threaded relationships with technical and operational stakeholders, maintains disciplined forecasting and account planning, and identifies opportunities to expand W&B usage across teams and business units. We ensure customers realize long-term value while contributing to sustained company growth.

About the Role:

As a Senior Account Manager, you will own the commercial strategy and outcomes for a portfolio of Enterprise and select Strategic customers. You will lead complex renewals, manage multi-threaded expansion and cross-sell motions, and scale W&B adoption across multiple teams and business units. You will run structured commercial workflows—including discovery, qualification, value and ROI articulation, pricing, negotiation, and forecasting—while maintaining detailed multi-year account plans that incorporate whitespace, risks, and growth sequencing. You will partner closely with Field Engineering to coordinate cross-functional deliverables, lead executive business reviews, and use adoption insights to influence expansion strategy. This role requires strong commercial judgment, technical curiosity, and the ability to operate autonomously in ambiguous, fast-moving environments.

Who You Are:

• 8+ years in B2B SaaS customer-facing roles commercializing technical products
• 5+ years in a quota-carrying Account Management, Sales, or Renewals role owning commercial outcomes
• Proven success managing Enterprise or Strategic customers with multi-threaded stakeholder engagement (practitioners → team leads → executives)
• Experience leading complex expansion, cross-sell, POV, and negotiation cycles
• Strong technical and product curiosity with familiarity in AI/ML, DevOps, or data infrastructure tooling
• Demonstrated ability to maintain detailed account plans including whitespace mapping, risk tracking, and multi-year growth planning
• Disciplined forecasting and CRM hygiene across renewal and expansion pipelines
• Able to self-direct in ambiguous environments and independently drive commercial outcomes
• Comfortable partnering with Field Engineering to connect technical adoption with business value and commercial strategy
  
  

Preferred:

• Experience collaborating with Field Engineering, Product, Engineering, or Support in technical SaaS
• Familiarity with Git, Python, Jupyter, PyTorch, Kubernetes, or cloud platforms (AWS, GCP, Azure)
• Experience supporting or selling to ML, MLOps, DevOps, or data infrastructure teams
• Background working with AI-native startups, foundation model builders, or Enterprise ML teams
  
  

Wondering if you’re a good fit?

We believe in investing in our people and value candidates who bring diverse experiences—even if you don’t meet every listed requirement. You may be a great fit if:

• You love building long-term customer partnerships and uncovering high-impact expansion opportunities
• You’re curious about how AI and ML teams build, scale, and productionize models—and you enjoy staying close to the latest advances in the field
• You’re an expert in navigating complex SaaS commercial cycles, managing multi-threaded accounts, and driving multi-team expansions

Why CoreWeave?

At CoreWeave, we work hard, have fun, and move fast!  We’re in an exciting stage of hyper-growth that you will not want to miss out on. We’re not afraid of a little chaos, and we’re constantly learning. Our team cares deeply about how we build our product and how we work together, which is represented through our core values:

• Be Curious at Your Core
• Act Like an Owner
• Empower Employees
• Deliver Best-in-Class Client Experiences
• Achieve More Together

We support and encourage an entrepreneurial outlook and independent thinking. We foster an environment that encourages collaboration and provides the opportunity to develop innovative solutions to complex problems. As we get set for takeoff, the growth opportunities within the organization are constantly expanding. You will be surrounded by some of the best talent in the industry, who will want to learn from you, too. Come join us!

The base salary range for this role is $165,000 to $189,000. The starting salary will be determined based on job-related knowledge, skills, experience, and market location. We strive for both market alignment and internal equity when determining compensation. In addition to base salary, our total rewards package includes a discretionary bonus, equity awards, and a comprehensive benefits program (all based on eligibility).

#LIjobs

What You’ll Do
The Field Engineering team at Weights & Biases plays a vital role in ensuring customer success and adoption of our platform. As part of this team, we partner with Sales, Support, Product, and Engineering to lead technical success after the sales process. We work closely with some of the most advanced AI teams in the world, helping them build, optimize, and scale their ML and GenAI workflows across industries such as computer vision, robotics, natural language processing, and large language models (LLMs).

About the Role

We’re hiring a Post-Sales AI Solutions Engineer (AISE), Scale Customer Success to help customers successfully implement and scale AI/ML workflows and GenAI/agentic applications on Weights & Biases. You will design and deliver technical enablement and adoption programs that reach many customers at once, create reusable assets that improve self-serve success, and use product signals and feedback loops to continuously improve outcomes.

Key Responsibilities 

1. Run 1-to-many onboarding and enablement programs
   Own and deliver scalable onboarding and adoption motions (webinars, cohort sessions, group training, and office hours) that help customers get to value quickly and consistently.
2. Build reusable technical assets that drive self-serve success
   Create and maintain playbooks, reference architectures, templates, sample code/notebooks, and troubleshooting guides that standardize best practices and reduce repeated 1:1 work.
3. Operate the scaled motion using signals and feedback loops
   Use product usage signals and customer patterns to segment and trigger the right interventions, track program impact (activation, time-to-first-value, feature adoption), and feed recurring insights back to Support, Product, and Field Engineering to continuously improve the scaled journey.

Who You Are

• 3–5 years of relevant experience in a similar role
• Strong programming proficiency in Python
• Experience with deep learning frameworks (TensorFlow/Keras, PyTorch Lightning) and tools (e.g., Streamlit, LangChain)
• Familiarity with cloud platforms (AWS, GCP, Azure)
• Excellent communication and presentation skills, both written and verbal
• Organized and outcomes-driven: you can run programs, measure impact, and iterate.

Preferred

• Experience with GenAI and LLMs
• Proficiency with HuggingFace, Fastai, scikit-learn, XGBoost, LightGBM, or Ray
• Experience with hyperparameter optimization solutions
• Background in data engineering, MLOps, or LLMOps, with tools such as Docker and Kubernetes
• Familiarity with data pipeline tools

Wondering if you’re a good fit?
We believe in investing in our people, and value candidates who can bring their own diversified experiences to our teams – even if you aren't a 100% skill or experience match. Here are a few qualities we’ve found compatible with our team. If some of this describes you, we’d love to talk.

• You love collaborating with top AI teams to solve real-world ML and GenAI challenges
• You’re curious about how to optimize AI workflows at scale
• You’re an expert in Python and ML tooling, with a passion for enabling customer success

Why Us?
We work hard, have fun, and move fast! We’re in an exciting stage of hyper-growth that you will not want to miss out on. We’re not afraid of a little chaos, and we’re constantly learning. Our team cares deeply about how we build our product and how we work together, which is represented through our core values:

• Be Curious at Your Core
• Act Like an Owner
• Empower Employees
• Deliver Best-in-Class Client Experiences
• Achieve More Together

We support and encourage an entrepreneurial outlook and independent thinking. We foster an environment that encourages collaboration and provides the opportunity to develop innovative solutions to complex problems. As we get set for takeoff, the growth opportunities within the organization are constantly expanding. You will be surrounded by some of the best talent in the industry, who will want to learn from you, too. Come join us!

The base salary range for this role is $139,000 to $204,000. The starting salary will be determined based on job-related knowledge, skills, experience, and market location. We strive for both market alignment and internal equity when determining compensation. In addition to base salary, our total rewards package includes a discretionary bonus, equity awards, and a comprehensive benefits program (all based on eligibility).

What You’ll Do:

The Account Manager owns the commercial and relationship aspects of the post-sales journey across a portfolio of Digital Native and select Enterprise customers. You will drive renewals, identify and close upsell and cross-sell opportunities, and ensure customers achieve measurable adoption outcomes with Weights & Biases (W&B). You’ll partner closely with Field Engineering (FE)—who leads technical success—while you lead the commercial motions including renewal execution, usage-to-value alignment, growth pipeline creation, and multi-threaded stakeholder engagement.

About the role:

You will own renewals and lead commercial execution across your assigned accounts, including identifying and closing upsell and cross-sell opportunities. You will partner closely with Field Engineering (FE) while running the commercial motions: renewal execution, usage-to-value alignment, growth pipeline creation, and multi-threaded stakeholder engagement. You will maintain detailed account plans (account maps, whitespace, usage trends, risks, and growth opportunities) and maintain weekly forecasting accuracy and CRM hygiene across renewals and growth motions. This role requires comfort engaging highly technical personas (ML engineers, researchers, PhDs) and operating with autonomy in a rapidly evolving AI ecosystem; it is not a playbook-driven role.

• Own renewals, upsells, and cross-sells across your assigned accounts.
• Build and maintain detailed account plans including account maps, whitespace, usage trends, risks, and growth opportunities.
• Generate growth pipeline by identifying new use cases, teams, and product opportunities within existing accounts.

Who You Are:

• Based in the Bay Area (San Francisco, Peninsula, South Bay) strongly preferred; secondary options: New York City or Northern NJ.
• 6+ years in B2B/SaaS customer-facing roles commercializing technical products.
• 3+ years in a quota-carrying Account Management, Sales, or Renewals role owning post-sales outcomes.
• Strong, genuine interest in AI/ML and the evolving machine learning ecosystem.
• High technical and product curiosity; comfortable speaking with developers, ML engineers, and researchers.
• Proven ability to drive growth motions (upsells, cross-sells) and manage retention in technical accounts.
• Strong operational discipline: forecasting accuracy, CRM hygiene, renewal cadence management.
• Thrives in ambiguity; able to self-direct and operate independently without detailed playbooks.

Preferred:

• Experience working with ML, MLOps, DevOps, or data infrastructure teams.
• Familiarity with Git, Jupyter, Python, PyTorch, or cloud platforms (AWS, GCP, Azure).
• Experience designing or executing account activation or expansion strategies.
• Exposure to AI-native companies, model builders, or generative AI workflows.

Wondering if you’re a good fit?

We believe in investing in our people, and value candidates who can bring their own diversified experiences to our teams – even if you aren’t a 100% skill or experience match. Here are a few qualities we’ve found compatible with our team. If some of this describes you, we’d love to talk.

• You love owning renewals and growth motions while building multi-threaded stakeholder relationships.
• You’re curious about AI/ML and enjoy learning quickly in a rapidly evolving ecosystem.
• You’re an expert in operating with strong discipline across account plans, forecasting, and CRM hygiene to drive commercial outcomes.

The base salary range for this role is $95,000 to $130,000. The starting salary will be determined based on job-related knowledge, skills, experience, and market location. We strive for both market alignment and internal equity when determining compensation. In addition to base salary, our total rewards package includes a discretionary bonus, equity awards, and a comprehensive benefits program (all based on eligibility).