MAS). Advanced Python programming skills, with a strong emphasis on writing efficient, scalable, and maintainable code. Proven experience with TensorFlow/PyTorch/Jax, Scikit-learn, and MLOps workflows for training, deployment, and monitoring of ML models. Experience working with Polars and/or Pandas for high-performance data More ❯
or a related field, or equivalent professional experience Strong programming skills in Python, with experience in machine learning frameworks such as PyTorch, TensorFlow or JAX Hands-on experience fine-tuning LLMs and optimising hyperparameters for improved performance Skilled in building LLM-powered applications for real-time decision-making and automation More ❯
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 More ❯
experience of Python or C++ Deep understanding of machine learning and statistics Strong knowledge of algorithm design and data structures Proven experience with TensorFlow, JAX, PyTorch, or similar leading deep learning frameworks Recent experience conducting applied research to improve the quality and training/serving efficiency of large transformer-based More ❯
language understanding The ability to write clean and maintainable code in Python, Rust, C++ or Java Experience with frameworks such as PyTorch, Tensorflow or JAX Strong written and verbal communication skills to document your work effectively and present it to relevant stake-holders and users A passion for identifying and More ❯
record in top-tier conferences (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR). Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ). Excellent communication skills to report and present research More ❯
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 More ❯
control management, build processes, testing, and operations experience - Bachelor's degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training. The group presents a lot of opportunity for optimization and scaling large deep learning More ❯
regression techniques, etc.) Familiarity with Deep Learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) Familiarity with PyTorch , TensorFlow , JAX or similar frameworks Strong coding abilities in Python and/or C++ Preferred Qualifications: 2D & 3D Generative AI Reinforcement Learning LLMs and Natural Language Processing More ❯
industry, including knowledge of the landscape and Pure’s solutions with a focus on the following: Deep understanding of industry AI frameworks (pytorch, tensorflow, jax etc) Deep understanding on high performance, low latency, highly parallelized compute solutions at scale (compute, network, storage) Deep understanding of accelerated compute environments (GPU, ASIC More ❯
data-driven role, with exposure to: 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; developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); iterating on network architectures and model More ❯
algos from scratch. In addition, the following would be an advantage: Experience with Large Language Models and notably Vision Language Models. Working knowledge of JAX/Pytorch or similar frameworks. Experience with data analysis at scale, data visualization and data ingestion pipelines for training models. A strong interest in Agents. More ❯
quality ML systems. Preferred Qualifications: Master's degree or PhD in Computer Science or a related technical field. Experience with modern ML frameworks (e.g., JAX, Pytorch or TensorFlow). Experience with launching applied Machine Learning/Natural Language Processing (ML/NLP) projects. About the Job Google's software engineers More ❯
of multiagent systems research in topics such as multiagent reinforcement learning, game theory, social choice, bandit algorithms. Experience with relevant ML frameworks such as JAX, TensorFlow, or PyTorch. In addition, the following would be an advantage: Experience with large language models, and/or AI agents. Familiarity with large-scale More ❯
and algorithm scaling, or neural network architectures. Experience communicating complex research for public audiences of peers. Experience with deep learning frameworks such as Pytorch, Jax, or Tensorflow. Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment. Preferred Qualifications Proven More ❯
machine learning, with 2+ years of hands-on experience in LLM fine-tuning, RAG systems, and prompt engineering. Proficiency with frameworks like PyTorch, Tensorflow, JAX, Pandas, and OpenCV. A solid understanding of transformer models and their common variants, loss functions, and pitfalls. A quick learner with a structured, organized approach More ❯
by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong More ❯
applications to engineering teams and C-Level executives. This requires deep familiarity across the stack - compute infrastructure (Amazon EC2, Amazon EKA), ML frameworks PyTorch, JAX, orchestration layers Kubernetes and Slurm, parallel computing (NCCL, MPI), MLOPs, through to Amazon SageMaker Hyperpod, Amazon Bedrock as well as target use cases in the More ❯
Machine Learning (ML), including modern deep learning methods. Ideally you will have experience in Python ML and Reinforcement Learning libraries such as Tensorflow, Keras, Jax, Pytorch Strong programming skills, preferably in python, KDB, ideally also Java or C++ Deep Experience in quantitative research, data analysis, strategy design and creation Educated More ❯
position such as post doctoral roles, a proven track record of publications, or contributions to machine learning codebases. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas Strong knowledge of linear algebra, calculus and statistics Experience working in a scientific More ❯
transformers, generative adversarial networks, and diffusion models. You are proficient with: Deep learning frameworks (Pytorch gives bonus points, alternatively other frameworks like Tensorflow or JAX), Python 3, Numpy, OpenCV, Docker & Kubernetes. You are able to pick up knowledge of the state-of-the-art of a specific ML topic quickly More ❯
position such as post doctoral roles, a proven track record of publications, or contributions to machine learning codebases. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas. Strong knowledge of linear algebra, calculus and statistics. Experience working in a scientific More ❯
of owning projects end to end Problem-solving mindset with the ability to acquire new skills as needed Proficiency with PyTorch (or TF/JAX), as well as NVIDIA GPU ecosystems and optimization stacks Highly metric-based Strong Python and C++ skills Bonus Qualifications Experience optimizing kernels with Triton or More ❯
performance environments is a strong plus Strong engineering skills, including Python, CUDA, or C++ Knowledge of machine learning frameworks such as PyTorch, TensorFlow, or JAX Proficiency in GPU programming for training and inference acceleration (e.g., CuDNN, TensorRT) Experience with distributed training for scaling ML workloads (e.g., Horovod, NCCL) Exposure to More ❯
to lead new research areas and multi-team cross-PA collaborations across geographical boundaries. Prior experience with production-scale machine learning projects and systems: JAX, TensorFlow, or similar machine learning platforms. What we offer At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities More ❯
