13 of 13 Google JAX Jobs

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play ...

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.Familiarity with current and emerging regulations and standards, such as the EU AI Act, the Digital Services Act, the Digital Markets Act, GDPR ...

Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research ...

engineers through technical leadership, code reviews and best practices. Building systems that balance performance, latency, reliability and scalability. Technology Python PyTorch and/or JAX GPU-based model training and inference Modern machine learning infrastructure What We're Looking For Commercial experience building and deploying machine learning models into ...

help shape engineering best practices. Balance research ambitions with practical engineering trade-offs to deliver value quickly. Technology Python PyTorch and/or JAX Large-scale GPU training and inference infrastructure Modern ML deployment pipelines Ideal Experience Proven experience delivering machine learning systems into production. Strong understanding of modern ...

Strong Machine Learning Engineering experience Deep knowledge of modern ML architectures (Transformers, GNNs, Diffusion Models, etc.) Excellent Python skills with PyTorch and/or JAX Experience building scalable production ML systems Someone who enjoys turning cutting-edge research into robust, high-performance software Bonus experience: Distributed GPU training (multi ...

silicon. What they’re looking for: First degree in relevant STEM subject 3+ years in industry building production ML systems Python + PyTorch/JAX expertise for training, inference, and model export Experience building data pipelines for complex scientific/engineering datasets Solid software engineering fundamentals - testing, CI/ ...

operating production ML systems Comfortable working across models, infrastructure and product Enjoy working in fast-moving, early-stage environments Tech Stack Python * PyTorch * JAX * LLMs * vLLM * Vector Databases * Modern Agent Frameworks Get in touch for more details - McGregor Boyall is an equal opportunity employer and do not discriminate ...

have built and shipped ML systems used by real users. Examples of relevant experience include: Model training and evaluation Production inference systems PyTorch or JAX GPU-based training and inference ML experimentation and iteration Monitoring and debugging production ML systems AI reliability and performance optimisation End-to-end ownership … production Independent ownership and execution Excellent problem-solving ability A bias towards shipping, measuring outcomes, and iterating quickly Technical Environment Python PyTorch/JAX GPU-based training and inference systems OpenAI, Anthropic and open-source models SQL/NoSQL Docker Kubernetes AWS, Azure or GCP Compensation Compensation is intentionally ...

problem-solving skills with the ability to work independently and collaboratively. Clear written and verbal communication skills. Familiarity with deep learning frameworks (TensorFlow, PyTorch, JAX) advantageous. Experience with optimisation, reinforcement learning, or LLMs a plus. Exposure to chip design or applied AI research desirable. Keywords: Research Engineer/Machine … Learning/AI/LLMs/Reinforcement Learning/Generative Models/Deep Learning/PyTorch/TensorFlow/JAX/Python/C++/Semiconductor/Chip Design/Optimisation/London/Cambourne/UK/Permanent If you are interested in this Research Engineer position, please ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud ...

hackajob is partnering directly with Google to hire for this role. Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond … mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers ...

systems Strong Python software engineering skills Experience training, fine-tuning or deploying modern machine learning models Strong experience with PyTorch and/or JAX Experience building scalable ML infrastructure and inference pipelines Comfortable owning projects end-to-end in fast-moving environments Previous mentoring or technical leadership experience would … advantageous Tech Stack Python * PyTorch * JAX * GPU Training * LLMs * Modern ML Infrastructure Get in touch for more details - McGregor Boyall is an equal opportunity employer and do not discriminate on any grounds. ...