CrawlJobs Logo

Generative AI - ML System Engineering

United States, Sunnyvale · Job Posted February 18, 2026
Apply Position
Job Link Share

Job Description

We are looking for Machine Learning Systems Engineers who can help us build the world's largest end-to-end 3D native machine learning systems. You will help us build our end to end ML framework dedicated for 3D, from pretraining, to finetuning, inferencing, etc. We expect a combination of strong hands on engineering skills, eagerness to learn new things, and thrives in a fast-paced, high-ownership environment.

Job Responsibility

  • Work closely with researchers to co-design the next frontier of 3D & Spatial AI
  • Build and debug on top of modern PyTorch, for maximum parallelism and efficiency, and build clean and intuitive training infrastructure for our in-house foundational models
  • Identifying bottlenecks and optimizing for high throughput & efficient distributed model training across hundreds to thousands of GPUs
  • Implementing and maintaining 3D specific custom operators in Triton or CUDA
  • Implementing and maintaining novel data-loading framework and libraries
  • Building efficient inference endpoints with complex multi-stage model pipelines
  • Optimizing models through compilation, fusion, quantization, etc

Requirements

  • Experience in machine learning or high performance graphics
  • Solid practical understanding of at least one machine learning framework (e.g. PyTorch, JAX)
  • Strong ability to write beautiful and maintainable code in Python and/or C++
  • Ability to learn fast and dive into new concepts or complex codebases
  • Performance and efficiency oriented mindset, with a strong interest in the tiniest detail
  • Strong communication skills for working in a globally distributed team

Nice to have

  • A strong passion to navigate through the PyTorch internals, with hands-on experience in areas like torch.compile , fully_shard (FSDP2) APIs
  • Experience with building Triton kernels
  • Experiences with large-scale distributed training, familiarity with modern parallelization techniques: DP, TP, CP, PP, zero redundancy optimizers, etc
  • Experience with diffusion models in 3D or video
  • Experience with low precision bf16 or fp8 training

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Similar Jobs for

Generative AI - ML System Engineering

8 matching positions

Ai-first Core It Software Engineering: Software, Ml & Data

This is a Unified Application for our AI-First IT Transformation portfolio. We r...
Location
Location
United States , Santa Clara
Salary
Salary:
Not provided
paloaltonetworks.com Logo
Palo Alto Networks
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 3-5+ years of experience in Software Engineering, Data Science, or Machine Learning (Staff level)
  • 6-8+ years (Senior Staff)
  • 8-12+ years (Principal level)
  • Expert-level server-side development (Python, Java, Go) OR deep expertise in statistical modeling, ML algorithms, and LLM fine-tuning
  • Direct experience with RAG architectures, LLM APIs, and Vector Databases (e.g., Pinecone, Milvus)
  • Hands-on experience with Kubernetes, CI/CD, and distributed systems for large-scale AI deployment
Job Responsibility
Job Responsibility
  • Lead the hands-on development of core Enterprise IT Business software leveraging AI components and LLM infrastructure with both traditional and Generative AI model deployment
  • Build and industrialize agentic AI systems and multi-agent frameworks, ensuring secure and effective use of GenAI technologies at the platform level
  • Design and implement robust foundational data pipelines, perform advanced statistical analysis, and develop new ML models to drive autonomous system behavior
  • Design large-scale, distributed AI/ML systems optimized for low latency, high throughput, and developer-friendliness (Inference optimization)
  • Establish evaluation frameworks to measure AI quality (accuracy, hallucination rates) and overall system reliability across the Enterprise AI Factory
  • Fulltime
Read More
Arrow Right

Ai Research Scientist —Generative Ai For Materials Discovery

Location
Location
United States , Redmond
Salary
Salary:
154000.00 - 217000.00 USD / Year
meta.com Logo
Meta
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
  • Ph.D. degree in Machine Learning, Computational Chemistry, Materials Science, Chemical Engineering, Physics, or a closely related technical field
  • 3+ years of research experience in generative modeling applied to molecular systems, crystal structures, or materials science (academic or industry)
  • Familiarity with large-scale molecular and crystal databases and data processing pipelines for chemical data
  • Demonstrated expertise in deep generative models — including diffusion models, flow matching / continuous normalizing flows, variational autoencoders, or autoregressive models — with applications to 3D molecular or crystal structure generation
  • Programming proficiency in Python with hands-on experience in PyTorch or JAX
  • proficiency in building, training, and evaluating large-scale deep learning models
  • Track record of first-author publications in top-tier ML or computational chemistry venues (e.g., NeurIPS, ICML, ICLR, JACS, Nature Computational Science, Digital Discovery)
  • Solid understanding of crystallography fundamentals— and molecular representations (molecular graphs, SMILES, 3D conformers)
Job Responsibility
Job Responsibility
  • Develop, train, and deploy generative models (diffusion models, flow matching, variational autoencoders, transformer-based architectures) for molecular and crystal structure generation, property-conditioned design, and crystal structure prediction (CSP)
  • Design and implement reinforcement learning and alignment strategies (e.g., physics-informed reward signals from machine-learned interatomic potentials) to steer generative models toward physically stable and synthesizable candidates
  • Build foundational models and scalable pretraining pipelines that unify generative and predictive learning across molecules and crystalline materials, handling both discrete atom types and continuous 3D geometries
  • Collaborate closely with computational chemists to integrate first-principles calculations (DFT, force fields), molecular dynamics simulations, and domain-specific constraints into generative workflows
  • Partner with AI agent scientists to embed generative molecular design capabilities into LLM-based multi-agent systems, enabling closed-loop autonomous experiment planning, candidate generation, and decision making
  • Curate, preprocess, and manage large-scale molecular and crystal structure datasets for model training and benchmarking
  • Establish rigorous evaluation frameworks — measuring validity, novelty, uniqueness, stability, and synthesizability of generated structures — and benchmark against state-of-the-art methods
  • Contribute to the architecture and roadmap of the autonomous materials-discovery platform, ensuring generative design modules interface seamlessly with robotic workcells, characterization instruments, and data infrastructure
What we offer
What we offer
  • bonus + equity + benefits
  • Fulltime
Read More
Arrow Right

Engineering Manager – Fullstack & AI Engineering

We are looking for an Engineering Manager – AI Engineering who combines people l...
Location
Location
India
Salary
Salary:
Not provided
Codvo AI
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 8–14+ years of engineering experience with 2–5+ years in team/people management
  • Strong hands-on programming expertise in Node.js
  • Strong hands-on programming expertise in Python
  • Strong hands-on programming expertise in React.js
  • Strong hands-on programming expertise in Angular
  • Experience delivering AI-enabled applications, generative AI features, or ML integrated systems
  • Solid understanding of microservices, distributed systems, API design, and cloud platforms (AWS/Azure/GCP)
  • Proven ability to lead engineering teams and deliver complex projects end-to-end
Job Responsibility
Job Responsibility
  • AI Engineering Leadership
  • Hands-On Technical Contribution
  • Delivery & Execution
  • People Leadership
  • Architecture & Technical Direction
  • Cross-Functional Collaboration
  • Fulltime
Read More
Arrow Right

Gen AI Engineering and Scaled AI Transformation

Location
Location
Canada , Mississauga
Salary
Salary:
145100.00 - 217700.00 USD / Year
https://www.citi.com/ Logo
Citi
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years of progressive experience in software engineering, ML, or AI platforms, with 5+ years leading senior engineers and architects
  • 3+ years of hands‑on experience deploying LLM‑based systems in production environments at enterprise scale
  • Demonstrated authority across commercial and open‑source LLM ecosystems (e.g., OpenAI, Anthropic, Google, Llama), including model selection, fine‑tuning, and hosting strategies
  • Proven ability to define enterprise-wide GenAI standards, reference architectures, and reusable accelerators
  • Demonstrated leadership in establishing prompt engineering standards and orchestration patterns
  • Experience optimizing latency, throughput, accuracy, and token cost across large‑scale GenAI workloads
  • Bachelor’s degree/University degree or equivalent experience
  • Master’s degree preferred
Job Responsibility
Job Responsibility
  • Acts as a senior technical authority on Large Language Models, including both commercial and open‑source ecosystems (OpenAI, Gemini, Claude, Llama)
  • Leads model selection and deployment strategy, balancing use‑case fit, data sensitivity, cost efficiency, latency, accuracy, and regulatory constraints
  • Guides decisions on hosted vs. private vs. fine‑tuned models, ensuring optimal trade‑offs between performance, control, and operational risk
  • Establishes enterprise standards for LLM lifecycle management, including upgrades, regression validation, and decommissioning
  • Demonstrates hands‑on leadership in building GenAI applications using LangChain, LangGraph, LlamaIndex, and Hugging Face, translating experimentation into production systems
  • Architects agentic and multi‑step workflows, enabling tool‑use, reasoning chains, state management, and orchestration at enterprise scale
  • Sets reusable reference patterns and accelerators for GenAI adoption across application teams
  • Ensures solutions are built with enterprise-grade reliability, explainability, and extensibility
  • Designs and delivers robust RAG architectures that ground GenAI outputs in trusted, auditable enterprise data
  • Leads implementation of vector databases and embedding strategies (pgvector, Pinecone, Weaviate, FAISS), aligned with data access and security models
  • Fulltime
Read More
Arrow Right

Director of AI Engineering

Mastercard is seeking a Director of AI Engineering to lead a specialised team of...
Location
Location
Ireland , Dublin 18
Salary
Salary:
Not provided
mastercard.com Logo
Mastercard
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Proven experience leading teams delivering AI/ML solutions in production, particularly in use-case driven environments
  • Strong hands-on understanding of transformer architectures and generative AI, including fine-tuning, prompting, embeddings, and evaluation
  • Experience bridging model development and real-world application, translating AI capabilities into business impact
  • Solid engineering background, with familiarity with Python, ML frameworks (e.g. PyTorch/TensorFlow), and production deployment patterns
  • Experience working with data engineering and MLOps practices to support training, evaluation, and inference at scale
  • Strong stakeholder management skills, with the ability to align technical delivery to business priorities and measurable outcomes
  • Demonstrated ability to lead and develop high-performing teams, providing technical direction, coaching, and delivery oversight
  • Comfortable operating in a fast-moving, evolving AI landscape with ambiguity and shifting priorities
  • Excellent communication skills, able to articulate complex AI concepts to both technical and non-technical audiences
Job Responsibility
Job Responsibility
  • Lead a team of AI engineers focused on foundation model development, fine-tuning, and optimisation, particularly transformer-based and generative AI systems
  • Drive the end-to-end delivery of AI use cases, from problem definition through model integration and production deployment
  • Partner with product and business stakeholders to identify and prioritise high-impact AI use cases, translating them into clear technical execution plans
  • Ensure effective use of foundation models across use cases, including prompting strategies, embeddings, fine-tuning, evaluation, and performance optimisation
  • Collaborate with data engineering and platform teams to ensure data readiness, model integration, and scalable deployment patterns
  • Establish best practices for model evaluation, experimentation, and continuous improvement, ensuring solutions are robust and measurable
  • Embed responsible AI practices, including model validation, bias considerations, and appropriate guardrails
  • Provide technical guidance and coaching to engineers, ensuring high standards in both AI development and software engineering practices
  • Track delivery progress, manage risks, and ensure timely, high-quality execution of use cases aligned to program priorities
  • Fulltime
Read More
Arrow Right

AI / ML Engineer/Lead

We are seeking a highly skilled and experienced Senior AI Engineer to lead the d...
Location
Location
India , Pune; Mumbai; Indore; Jaipur
Salary
Salary:
Not provided
Codvo AI
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s or Master’s in Computer Science, Artificial Intelligence, Machine Learning, or related field
  • 5+ years of experience in AI/ML engineering with end-to-end pipeline development
  • Hands-on experience building and deploying LLM/RAG systems in production
  • Strong experience with public cloud platforms (AWS, Azure, or GCP)
  • Proficient in Python and libraries such as Transformers, SentenceTransformers, PyTorch
  • Deep understanding of GenAI infrastructure, LLM APIs, and toolchains like LangChain/LangGraph
  • Experience with RESTful API development and version control using Git
  • Knowledge of vector DBs (Qdrant, FAISS, Weaviate) and similarity-based retrieval
  • Familiarity with Docker, Kubernetes, and scalable microservice design
  • Experience with observability tools like Prometheus, Grafana, or Langfuse
Job Responsibility
Job Responsibility
  • Generative AI Pipeline Development
  • Data and Document Ingestion
  • AI Service Integration
  • Retrieval-Augmented Generation (RAG) Pipelines
  • LLM Integration and Optimization
  • Backend Services Ownership
  • Fulltime
Read More
Arrow Right

Ai Engineering Lead

At Aden, artificial intelligence is not just a feature; it is the core driver of...
Location
Location
United States , San Francisco
Salary
Salary:
250000.00 - 350000.00 USD / Year
helpcare.ai Logo
Helpcare AI
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 4+ years of software engineering experience, with at least 2 years dedicated to deploying machine learning models, LLMs, or complex AI systems in production environments
  • Expert-level proficiency in Python
  • Deep understanding of ML frameworks (PyTorch, TensorFlow) and LLM orchestration tools (LangChain, LlamaIndex, or proprietary/custom frameworks)
  • Strong background in distributed systems, vector databases (Pinecone, Weaviate, Milvus), and cloud infrastructure (AWS/GCP)
  • Solid grasp of transformer architectures, fine-tuning methodologies (LoRA, QLoRA), and prompting strategies for complex reasoning tasks
  • A relentless focus on security, accuracy, and edge-case handling
Job Responsibility
Job Responsibility
  • Model Deployment & Optimization: Design, deploy, and maintain state-of-the-art foundation models and custom-trained AI systems within a high-throughput, low-latency B2B enterprise environment
  • Agentic Framework Development: Architect and implement robust, multi-agent workflows capable of handling complex operations, from risk assessment and fraud detection to automated reconciliation and dynamic routing
  • RAG & Data Pipelines: Build and optimize Retrieval-Augmented Generation (RAG) pipelines that ingest, vectorize, and retrieve massive volumes of structured and unstructured data
  • Evaluation & Alignment: Develop rigorous, automated testing frameworks to evaluate model performance, specifically focusing on hallucination reduction, mathematical accuracy, and strict adherence to compliance standards
  • Infrastructure Integration: Work closely with backend engineers to integrate AI services seamlessly into our core Rust/Go/Python-based APIs
  • Continuous Improvement: Stay at the absolute bleeding edge of AI research
What we offer
What we offer
  • Competitive salary and substantial equity package
  • Comprehensive health, dental, and vision insurance
  • The opportunity to build the underlying architecture for the AI economy from the ground up
  • Flexible, remote-friendly work culture with regular team offsites
  • Fulltime
Read More
Arrow Right

Internship AI Engineering

Location
Location
Poland , Lodz
Salary
Salary:
Not provided
ventionteams.com Logo
Vention
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Proficient in Python syntax, collections, pip, and the fundamentals of OOP
  • Good understanding of basic linear algebra, including vector and matrix operations, derivatives, and function plots
  • Knowledge of machine learning fundamentals such as overfitting, gradient descent, classification, regression, metrics, and examples of common ML models
  • Familiarity with Pandas for data processing and analysis
  • Good grasp of SQL and database schema design
  • Theoretical understanding of NLP fundamentals, transformer models and prompt engineering techniques
  • English proficiency at B2 level or higher
  • Capacity to study for 6 hours daily
Job Responsibility
Job Responsibility
  • Design, develop, and deploy Retrieval-Augmented Generation (RAG) and computer vision systems for real business use cases
  • Evaluate model output quality and system performance using quantitative metrics and structured experiments
  • iterate to improve accuracy, latency, and reliability
  • Validate technical hypotheses through prototyping and document findings
  • Collaborate closely with teammates using agile practices (standups, sprint planning, reviews), present demos and results to internal stakeholders
What we offer
What we offer
  • Enjoy personalized learning with intimate group sizes of 3-15 or opt for a one-on-one experience
  • Our dynamic curriculum offers a mix of hands-on practice and essential theory, tailored for groups or adjusted to fit individual needs
  • Give yourself at least three months to dive deep into the material in a group, or choose an individual internship length that aligns with international standards
  • Discover the industry inside out. This internship provides insights into the IT world, giving you a leg up in your future career
  • Receive guidance and support from an experienced mentor throughout your internship journey
  • Beyond learning, there's a chance for employment. Successful interns might land a full-time job with us after the program
  • Dive into real-world projects! Get hands-on experience with genuine IT challenges and see firsthand the solutions in action
  • Parttime
Read More
Arrow Right