P-984

Founded in late 2020 by a small group of machine learning engineers and researchers, Mosaic AI enables companies to securely

fine-tune, train and deploy custom AI models on their own data, for maximum security and control. Compatible with all major cloud

providers, the Mosaic AI platform provides maximum flexibility for AI development. Introduced in 2023, Mosaic AI’s pretrained

transformer models have established a new standard for open source, commercially usable LLMs and have been downloaded over 3

million times. 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.

Now part of Databricks since July 2023, we are passionate about enabling our customers to solve the world's toughest problems —

from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by

building and running the world's best data and AI platform so our customers can use deep data insights to improve their business.

We leap at every opportunity to solve technical challenges, striving to empower our customers with the best data and AI

capabilities.

Summary:

Databricks Mosaic AI is hiring experienced machine learning platform engineers to build out our customer-facing generative AI

platform for the ML development lifecycle including data generation, training, evaluation, serving, and agent-building.

You will thrive in this role if you have a strong sense of end-to-end ownership and enjoy translating user requirements into

product interfaces and building the backend distributed systems to power those interfaces. In this role, you will have the

opportunity to contribute to all areas of our stack spanning from user-facing features to low-level GPU orchestration.

You will:

Play a key role in the end-to-end design and implementation of our product which is a platform for powering use cases across

training and serving of generative AI models

Work closely with both customers and internal ML researchers to identify key areas of development for our generative AI

platform

Have strong end-to-end product ownership, translating product requirements into user interfaces and backend distributed system

design and own end-to-end implementation

Design and build the core platform infrastructure that supports our customer-facing product features
Ensure the reliability, security, and scalability of the backend distributed systems that power all aspects of our product

We look for:

4+ years of hands-on programming experience with at least one modern language such as Python, Scala, Go, or C++
Strong sense of distributed systems design and experience building large-scale systems
Experience building ML platform systems for applications in the ML model development lifecycle such as data preparation, model

training, model evaluation, and model serving

Direct experience developing ML models is a plus but not required
Strong sense of end-to-end product ownership as well as intuition for both robust system design and product usability
Effective communication skills and the ability to articulate complex technical ideas to cross-disciplinary internal and

external stakeholders.

We value candidates who are curious about all parts of the company's success and are willing to learn new technologies along the

way.

Pay Range Transparency

Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and

represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles. Actual

compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related

skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above,

Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include

eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your

location is in visit our page here.

Local Pay Range

$166,000—$225,000 USD

About Databricks

Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and

over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI.

Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse,

Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on Twitter, LinkedIn and Facebook.

Benefits

At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific

details on the benefits offered in your region click here.

Our Commitment to Diversity and Inclusion

At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to

ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment

at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or

expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation,

socio-economic status, veteran status, and other protected characteristics.

Compliance

If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's

discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an

applicant on this basis alone.

Role

Senior Machine Learning Engineer - GenAI Platform

Similar Jobs

Staff Machine Learning Engineer

Databricks•San Francisco

P-1504 The Applied AI team at Databricks sits at the forefront of advancing GenAI-powered products. Over the past years, we’ve launched Databricks Assistant, AI/BI Genie, and Agent Bricks working with product teams, and made significant strides in LLM quality for these products. These products are used by 100s of thousands of Databricks users every day. We are tackling challenging problems like code suggestion, error detection and correction, text-to-sql generation, automatic pipeline generation, knowledge QA and many others. As our GenAI products continue to evolve, we are seeking multiple GenAI Engineers from junior levels to more senior levels to drive the next phase of development. In 2025, we will focus on enhancing LLM quality, expanding GenAI capabilities across Databricks products, and strengthening our platform architecture to enable seamless AI interactions at scale. Key Responsibilities * Shape the direction of our applied AI areas and intelligence features in our products. Drive the development and deployment of state-of-the-art AI models and systems that directly impact the capabilities and performance of Databricks' products and services (e.g., Databricks Assistant and AI/BI Genie). * Develop novel data collection, fine-tuning, and LLM technologies that achieve optimal performance on specific tasks and domains. * Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and model evaluation, enabling rapid experimentation and iteration. * Work closely with cross-functional teams, including AI researchers, ML engineers, and product teams, to deliver impactful AI solutions that enhance user productivity and satisfaction. * Build scalable, reusable backend systems to support GenAI products across the company. Develop robust logging, telemetry, and evaluation harnesses to ensure reliable model performance. What We’re Looking For * 2-8 years of machine learning engineering experience in high-velocity, high-growth companies. Alternatively, a strong background in relevant ML research in academia will be considered as an equivalent qualification. * Strong track record of working with language modeling technologies. This could include the following: Developing generative and embedding techniques, modern model architectures, fine tuning / pre-training datasets, and evaluation benchmarks. * Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures. * Ability to drive end-to-end model development, from research and prototyping to deployment and monitoring. * Strong analytical and problem-solving skills, with a passion for improving AI-driven user experiences. * Strong coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews and deployment. * Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a bonus. Why Join Us? At Databricks, we are building state-of-the-art AI solutions that redefine how users interact with data and our products. You’ll have the opportunity to shape the future of AI-driven products at Databricks, work with cutting-edge models, and collaborate with a world-class team of AI and ML experts. If you're excited about pushing the boundaries of AI in real-world applications, we’d love to hear from you! Please note we are open to employees working from our Mountain View, CA office for this position. Pay Range Transparency Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles. Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here. Local Pay Range $190,000—$285,000 USD About Databricks Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on Twitter, LinkedIn and Facebook. Benefits At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here. Our Commitment to Diversity and Inclusion At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics. Compliance If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

Deadline: Aug 16Posted: Jul 17

Senior Software Engineer, AI

Okta•San Francisco

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta’s Business Technology organization builds secure and intelligent internal platforms that power our global workforce. Our AI & Automation team is delivering next-generation tools and experiences by integrating GenAI and intelligent automation into workflows across IT, HR, Finance, Sales, Marketing and Customer Support. We focus on real-world applications: virtual agents, AI copilots, internal RAG services, and AI-augmented self-service portals — all with scale, governance, and user experience in mind. ---------------------------------------------------------------------------------------------------------------------------------- THE OPPORTUNITY As a Senior Software Engineer, Applied AI, you’ll play a key role in building user-facing and backend systems that leverage GenAI to improve internal experiences and operations. This role requires strong full-stack engineering skills, with an emphasis on both AI integration and building intuitive, performant UIs that make AI accessible and useful to our internal customers. You’ll work closely with software engineers, product managers, and designers to build secure, intelligent tools for employees across Okta. ---------------------------------------------------------------------------------------------------------------------------------- WHAT YOU’LL DO * Design and build end-to-end GenAI-powered applications, including web-based UIs, API services, and backend orchestration. * Implement and integrate LLM-based experiences using frameworks like LangChain, LlamaIndex, and tools like OpenAI, Claude, or Gemini. * Define, implement, and champion operational excellence standards (SLOs, observability, incident response frameworks) for all services deployed. * Develop responsive, accessible, and modern frontend interfaces using frameworks like React or Vue — with a focus on usability, performance, and trust in AI outputs. * Build and maintain a library of reusable frontend components and hooks that allow other business delivery teams to easily "drop in" GenAI capabilities into their own applications. * Build and maintain retrieval-augmented generation (RAG) pipelines with vector search and embedding strategies (e.g., Pinecone, FAISS, Qdrant). * Collaborate with designers and product managers to rapidly iterate on UX patterns for AI-powered experiences (e.g., prompt inputs, citations, summaries). * Ensure security, privacy, observability, and test coverage across the full stack. * Contribute to architecture decisions, engineering standards, and best practices for AI/automation systems. * Partner with platform and infrastructure teams to ensure services scale reliably across the org. ---------------------------------------------------------------------------------------------------------------------------------- WHAT YOU’LL BRING * 5–8 years of software engineering experience with full-stack development, including 2+ years of building AI/ML-driven applications. * 5+ years of experience in frontend technologies like React, TypeScript, or Vue, and comfort working on UI/UX for internal tools or enterprise applications. * Strong Python development skills and 2+ years experience building cloud-based services using AWS, Docker, and RESTful APIs. * Hands-on experience with LLM integration, RAG pipelines, prompt engineering, or orchestration frameworks like LangChain or LlamaIndex. * Strong background in distributed systems, APIs, microservices, container orchestration (ECS/EKS), and cloud platforms (AWS/GCP/Azure). * Familiarity with secure coding, authentication/authorization, and internal data governance best practices. * Ability to collaborate across engineering, design, and product teams — with a strong sense of user empathy and technical ownership. * Bonus: Exposure to design systems, AI evaluation tooling, or real-time application performance monitoring. ---------------------------------------------------------------------------------------------------------------------------------- WHY JOIN OKTA * Make AI Real: Design and build AI-powered apps used daily by Okta employees. * Full-Stack Challenge: Tackle end-to-end problems — from LLM orchestration to intuitive UIs. * Trusted Innovation: Join a team committed to security, ethics, and technical excellence in AI. #LI-MK1 #LI-hybrid P24738_3354842 The annual base salary range for this position for candidates located in the San Francisco Bay area is between: $165,000—$247,000 USD Below is the annual base salary range for candidates located in California (excluding San Francisco Bay Area), Colorado, Illinois, New York and Washington. Your actual base salary will depend on factors such as your skills, qualifications, experience, and work location. In addition, Okta offers equity (where applicable), bonus, and benefits, including health, dental and vision insurance, 401(k), flexible spending account, and paid leave (including PTO and parental leave) in accordance with our applicable plans and policies. To learn more about our Total Rewards program please visit: https://rewards.okta.com/us. The annual base salary range for this position for candidates located in California (excluding San Francisco Bay Area), Colorado, Illinois, New York, and Washington is between: $147,000—$221,000 USD The Okta Experience * Supporting Your Well-Being * Driving Social Impact * Developing Talent and Fostering Connection + Community We are intentional about connection. Our global community, spanning over 20 offices worldwide, is united by a drive to innovate. Your journey begins with an immersive, in-person onboarding experience designed to accelerate your impact and connect you to our mission and team from day one. Okta is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, marital status, age, physical or mental disability, or status as a protected veteran. We also consider for employment qualified applicants with arrest and convictions records, consistent with applicable laws. If reasonable accommodation is needed to complete any part of the job application, interview process, or onboarding please use this Form to request an accommodation. Notice for New York City Applicants & Employees: Okta may use Automated Employment Decision Tools (AEDT), as defined by New York City Local Law 144, that use artificial intelligence, machine learning, or other automated processes to assist in our recruitment and hiring process. In accordance with NYC Local Law 144, if you are an applicant or employee residing in New York City, please click here to view our full NYC AEDT Notice.

Deadline: Aug 16Posted: Jul 15

Senior Machine Learning Engineer

Taskrabbit•San Francisco

ABOUT TASKRABBIT: Taskrabbit is a marketplace platform that conveniently connects people with Taskers to handle everyday home to-do’s, such as furniture assembly, handyman work, moving help, and much more. At Taskrabbit, we want to transform lives one task at a time. As a company we celebrate innovation, inclusion and hard work. Our culture is collaborative, pragmatic, and fast-paced. We’re looking for talented, entrepreneurially minded and data-driven people who also have a passion for helping people do what they love. Together with IKEA, we’re creating more opportunities for people to earn a consistent, meaningful income on their own terms by building lasting relationships with clients in communities around the world. Taskrabbit is a hybrid company with employees distributed across the US and EU and a Built In — Best Places to Work (2022, 2023, 2024, 2025) continually ranked across multiple national and regional categories. Join us at Taskrabbit, where your work will be meaningful, your ideas valued, and your potential unleashed! We are not able to provide visa sponsorship (including H-1B, OPT, or other employment-based visas) for this position. Candidates must be legally authorized to work in the United States without employer sponsorship now or in the future. ABOUT THE ROLE Machine Learning is a cornerstone at Taskrabbit, and we're looking for a seasoned Senior Machine Learning Engineer to join our team and help shape the future of ML/AI at Taskrabbit. This is a unique, full-stack role for an individual who is passionate about the entire machine learning lifecycle—from initial research and model development to building the robust infrastructure required to deploy and scale your work. As a Senior Machine Learning Engineer, you will tackle exciting challenges that directly impact how people discover and connect with home services on the Taskrabbit platform. You will play a crucial role in advancing our capabilities in areas like search ranking, content discovery, and recommender systems. You will collaborate closely with data scientists and other engineers to design and implement novel algorithms, and you will partner with software engineers to ensure the scalability, reliability, and optimization of our models in production. WHAT YOU'LL WORK ON: * Model Development & Research: Research, design, and implement machine learning models to solve key business problems in areas like search ranking, recommendations, and content discovery. * End-to-End ML Lifecycle: Own the entire lifecycle of ML models, including feature engineering, training, evaluation, deployment, and monitoring. * Infrastructure & Scalability: Build scalable and reliable ML infrastructure and data pipelines that support reproducible feature engineering and machine learning model deployment in real-time, near real-time, and batch processes. * Performance & Quality: Build monitoring services to understand data quality and model performance of complex systems, and collaborate with engineering and science teams to optimize existing algorithms for training and evaluation. * Software Engineering Excellence: Independently solve complex problems, write clean, efficient, and sustainable code, and actively participate in code reviews, documentation, and the full software engineering lifecycle. YOUR AREAS OF EXPERTISE: We welcome applicants from a variety of backgrounds and experiences. Below gives you a sense of how we're thinking about what you'll need to be successful in the role. * BS, MS, or PhD in Computer Science, Statistics, Operations Research, or a related quantitative field. * 5+ years of industry experience building and deploying high-quality, production-grade machine learning models and systems. * Strong theoretical knowledge and hands-on experience in machine learning, particularly in areas like search, ranking, recommender systems, or NLP. * Solid software engineering skills with proficiency in one or more programming languages, including Python. The candidate should have experience with popular ML libraries like Scikit-learn, lightgbm, xgboost, TensorFlow, PyTorch, etc. * Proficiency in SQL is also required for writing complex queries and transforming data. * Experience building REST API-based services. * Experience with modern data and ML technologies, such as Docker, Kubernetes, Kafka, Airflow, data warehouses (eg snowflake, redshift or BigQuery), and data lakes. * Familiarity with dbt (Data Build Tool) is a plus for transforming and testing data. * Familiarity with tools for Infrastructure as Code, such as Terraform, and CI/CD pipelines. * Excellent communication skills, with the ability to present complex findings and recommendations clearly to both technical and non-technical audiences. * A passion for quickly learning new technologies and a drive to solve challenging problems. COMPENSATION & BENEFITS: At Taskrabbit, our approach to compensation is designed to be competitive, transparent, and equitable. Total compensation consists of base pay + bonus + benefits + perks. The base pay range for this position is $150,000 - $200,000. This range is representative of base pay only, and does not include any other total cash compensation amounts, such as company bonus or benefits. Final offer amounts may vary from the amounts listed above and will be determined by factors including, but not limited to, relevant experience, qualifications, geography, and level. YOU’LL LOVE WORKING HERE BECAUSE: * Taskrabbit is a Hybrid Company. We value flexibility and choice but also stay committed to regular in-person connection. * The People. You will be surrounded by some of the most talented, supportive, smart, and kind leaders and teams -- people you can be proud to work with! * The Diverse Culture. We believe that we make better decisions when our workforce reflects the diversity of the communities in which we operate. Women make up half of our leadership team and our diversity representation is above that of the tech industry average. * The Perks. Taskrabbit offers our employees with employer-paid health insurance and a 401k match with immediate vesting for our US based employees. We offer all of our global employees generous and flexible time off with 2 company-wide closure weeks, Taskrabbit product stipends, wellness + productivity + education stipends, IKEA discounts, reproductive health support, and more. Benefits vary by country of employment. TASKRABBIT’S COMMITMENT TO DIVERSITY AND INCLUSION: An Active Commitment to Equity within our Company and Platform. We are an inclusive community where all who share our mission and values belong. Our diverse team represents the communities we serve, breaking down systemic barriers, and transforming lives- one action at a time. Taskrabbit is an equal opportunity employer and values diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, ancestry, citizenship, sex, gender, gender identity, sexual orientation, age, marital status, military/veteran status, or disability status. Taskrabbit is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities. Taskrabbit will consider for employment all qualified applicants with criminal histories in a manner consistent with applicable law.

Deadline: Aug 16Posted: Jun 25

PythonDockerKubernetes