76 to 100 of 247 Permanent Google JAX Jobs

focus on building graph-based neural network (GNN) models, generating ScaNN-based embeddings, and training scalable ML models for search, recommendation, and classification systems. You will collaborate closely with Google Cloud engineers, architects, and data scientists to deliver innovative, production-ready AI solutions. Key Responsibilities Design and implement Graph Neural Network (GNN) architectures for enterprise-scale applications. Develop and … optimize vector embedding generation pipelines using ScaNN or similar ANN techniques. Train, fine-tune, and deploy ML/DL models using TensorFlow, PyTorch, JAX, or similar frameworks. Collaborate with data engineers and solution architects to integrate models into scalable cloud solutions. Perform model evaluations, A/B testing, and hyperparameter tuning for optimal performance. Build reusable pipelines and tools … documentation and presentations for internal and customer-facing stakeholders. Required Qualifications Bachelor’s degree in computer science, Mathematics or a related technical field or equivalent practical experience. Certifications Minimum: Google Professional Data Engineer Preferred: AWS Machine Learning Specialty Certification 7+ years in a customer facing role working with enterprise clients 4+ years of experience working in enterprise data warehouse More ❯

for enterprise-scale applications. Develop and optimize vector embedding generation pipelines using ScaNN or similar ANN techniques. Train, fine-tune, and deploy ML/DL models using TensorFlow, PyTorch, JAX, or similar frameworks. Collaborate with data engineers and solution architects to integrate models into scalable cloud solutions. Perform model evaluations, A/B testing, and hyperparameter tuning for optimal … technical documentation and presentations for internal and customer-facing stakeholders. Required Qualifications: Bachelors degree in computer science, Mathematics or a related technical field or equivalent practical experience. Certifications Minimum: Google Professional Data Engineer Preferred: AWS Machine Learning Specialty Certification 7+ years in a customer facing role working with enterprise clients 4+ years of experience working in enterprise data warehouse … Neural Networks (GNNs) using frameworks like DGL, PyTorch Geometric, or similar. Experience with ScaNN or other approximate nearest neighbor (ANN) techniques for vector similarity search. Hands-on experience with Google Cloud Platform (GCP) tools such as Vertex AI, BigQuery, and Dataflow. Strong problem-solving and communication skills, including the ability to work with clients and cross-functional teams. Preferred More ❯

supervised models in a Bayesian Optimization context (Gaussian processes, Bayesian Neural Networks) on small and large datasets. Strong experience in at least one ML framework (PyTorch/TensorFlow/Jax) and robust experience in Python data science ecosystem (Numpy, SciPy, Pandas, etc.) Experience using a cloud computing service to reduce runtime to train and evaluate deep learning models PhD More ❯

concepts and best practices (e.g., versioning, testing, CI/CD, API design, MLOps) Building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., PyTorch, MLFlow, JAX) Distributed computing frameworks (e.g., Spark, Dask) Cloud platforms (e.g., AWS, Azure, GCP) and HP computing Containerization and orchestration (Docker, Kubernetes) Ability to scope and effectively deliver projects Strong problem More ❯

s or PhD degree in a quantitative field (e.g., Computer Science, Machine Learning, Statistics, Physics, Mathematics) or equivalent industry experience. Strong background in deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Experience with distributed computing platforms (AWS, GCP, Azure, or on-prem clusters). Ideal: Kubernetes and Docker experience for scalable, reproducible workflows. Working at Lila Sciences, you would More ❯

Triton); cloud computing (on hyper-scaler platforms, e.g., AWS, Azure, GCP); building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; C/C++ for computer vision, geometry processing, or scientific computing; software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API More ❯

position, the following are important: Technical Ability A deep understanding of data science, analytics platforms/ML domains, especially GenAI and NLP. Experience with frameworks (eg. PyTorch, TensorFlow, or JAX) and LLM’s (e.g. prompt engineering, fine tuning, RAG). Experience building and fine-tuning various ML Models (supervised and unsupervised models, classification, clustering, regression etc) Hands on experience More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

AI-driven agent systems. This role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search capabilities (e.g., ScaNN, FAISS) for … AI agent frameworks (e.g., Dialogflow, LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g. … TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning algorithms. • Familiarity with MLOps principles and tools for deploying, managing, and monitoring ML models in More ❯

AI-driven agent systems. This role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search capabilities (e.g., ScaNN, FAISS) for … AI agent frameworks (e.g., Dialogflow, LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g. … TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning algorithms. • Familiarity with MLOps principles and tools for deploying, managing, and monitoring ML models in More ❯

AI-driven agent systems. This role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search capabilities (e.g., ScaNN, FAISS) for … AI agent frameworks (e.g., Dialogflow, LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g. … TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning algorithms. • Familiarity with MLOps principles and tools for deploying, managing, and monitoring ML models in More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to More ❯

Join to apply for the AI Research Scientist (AEC) role at Autodesk Continue with Google Continue with Google 2 weeks ago Be among the first 25 applicants Join to apply for the AI Research Scientist (AEC) role at Autodesk Direct message the job poster from Autodesk Principal Engineering Recruiter Product Software at Autodesk. I am an Autodesker." Job … and deploying production-ready solutions) Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry-specific data formats (e.g., BIM More ❯

as an architect, or technical leadership position Hands-on experience developing FPGA hardware solutions and deployment at scale. Familiarity with deep learning frameworks and tensor libraries like PyTorch, Tensorflow, JAX/FLAX Preferred Qualifications: BS and 8+ years of relevant work experience, MS and 7+ years of relevant work experience, or PhD and 4+ years of relevant work experience … 2+ years of experience in hardware acceleration of AI/ML models Experience in deep learning frameworks and tensor libraries like PyTorch, Tensorflow, JAX/FLAX Familiarity with building ML applications, LLM serving, GPU serving. Familiarity with containers and container orchestration systems Outstanding interpersonal communication skills (including listening, speaking, and writing) and ability to work well in a diverse More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to More ❯

have: •Experience in optimization, reinforcement learning, and/or large language models (LLMs) •Experience in chip design and applied AI research •Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Why Join Us •Work on groundbreaking AI research with real-world applications and a strong commitment to open-source contributions and scientific advancement •Collaborate with a team of world More ❯

self-motivated and independent worker. Strong English writing and verbal communication skills. Preferred Skills Proficiency in Python. Expertise in one of the standard machine learning frameworks, such as PyTorch, JAX, or TensorFlow. Broad knowledge of and practical experience with modern machine learning and optimization techniques. Excellent analytical, numerical, and modeling skills. Broad knowledge of photonics and general physics. Familiarity More ❯

Python, R, C++, Rust or Java, and their associated security considerations. Coding experience with AI & ML, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn. Personal Attributes Drive and lead continuous improvement with energy and passion Displays high levels of self-awareness and emotional intelligence in all situations Strong inter-personal More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of More ❯

Join to apply for the Machine Learning Engineer role at Graphcore Join to apply for the Machine Learning Engineer role at Graphcore Get AI-powered advice on this job and more exclusive features. As a Machine Learning Engineer in the More ❯