51 to 70 of 70 Permanent Google JAX Jobs

better outcomes, 80% less admin work Raised a $30M Series A from top investors Built a team of 40+ AI experts from Tesla, Google DeepMind, NVIDIA, and Databricks At Brain Co., your work will be deployed in the real world, not stuck in research. We move fast, with more … high level familiarity with the architecture and operation of large language models. Have personally implemented models in common ML frameworks such as PyTorch, Jax or TensorFlow. Possess a strong foundation in data structures, algorithms, and software engineering principles. Exhibit excellent problem‐solving and analytical skills, with a proactive approach ...

NAACL, ICLR) with specific focus on agent systems, tool use, or multi‐agent coordination. 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 … source libraries or writing production code. Experience training large‐scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud. You will enjoy Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting‐edge methods ...

NVLink/InfiniBand), Kubernetes/Slurm, and OS internals Expertise in Python and deep experience with modern deep learning frameworks (PyTorch and/or JAX) Advanced degree (MS or PhD) in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience at a frontier … orientation, and excellent collaborative communication Reliable and predictable availability Bonus Points Expertise in: CUDA kernel development, Triton/Pallas/CuTe DSLs, PyTorch/JAX internals, XLA optimization, or hardware acceleration (FPGA/ASIC) Knowledge of reinforcement learning, post-training, or fine-tuning techniques Knowledge of financial markets ...

much as possible on-device to enable low-power, accurate, and private AI products. This is where Plumerai leads, demonstrating better accuracy even than Google Nest. We are now starting to invest heavily in market adoption. We build the most accurate and efficient AI solutions by vertically integrating … backed by world-class investors such as Tony Fadell (creator of iPod, iPhone; founder of Nest), Hermann Hauser (founder of Arm), Zoubin Ghahramani (Google DeepMind), and others. Learn more here: 👉 Read more: TechCrunch, Series A funding announcement Role description We are looking for a Senior AI Research Engineer that ...

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

About WRITER WRITER is where the world’s leading enterprises orchestrate AI-powered work. Our vision is to expand human capacity through superintelligence. And we’re proving it is possible – through powerful, trustworthy AI that ...

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced DriverTM—to improve access to mobility … generative models. Strong publication record or history of impactful project delivery in RL or related areas. Proficiency in Python and standard ML frameworks (e.g., JAX, TensorFlow). Experience with large‐scale distributed training and data processing. Proven ability to lead complex and ambiguous technical projects from conception to completion. ...

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

We are supporting a pioneering technology organisation seeking a Senior Research Engineer to lead complex AI research and engineering initiatives. This position sits at the intersection of advanced research and practical engineering. You will help ...

DeepMind • London, UKBack to jobs1. Home2. Jobs3. UK4. Research Scientist, Agent Post-Training5h agoD## Research Scientist, Agent Post-TrainingDeepMind00ResearchFull-timeLondon, UKApply Job### Job DescriptionAt Google DeepMind, we've built a unique culture and work environment where long-term ambitious research can flourish. In this role, you will push … reinforcement learning. You will value various experience and backgrounds to create extraordinary impact.Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high ...

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

building defenses against risks, contributing to DeepMind's Frontier Safety Framework commitments. Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high … advance research in automated red-teaming and adversarial robustness, leveraging multi-turn/agentic attacks to systematically test for and uncover misuse vulnerabilities. Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users ...

hackajob is collaborating with Google to connect them with exceptional professionals for this role. At Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas may come from internal projects as well … related technical field, or equivalent practical experience. 2 years of experience with software engineering involving Python-based ML/scientific libraries such as JAX or internship experience in these areas or Open Source Software (OSS) contributions. 1 year of experience with machine learning. Preferred Qualifications Experience with large-scale ...

make that hardware serve large AI models at rack scale by building and extending the runtime and serving stack that connects PyTorch and JAX down to the metal. This is a whole-stack systems role. You’ll work where the runtime, the network, and the accelerator meet, partnering closely … cache and memory management, and reliability across distributed failure domains at cluster scale. Fluency at the framework boundary, connecting accelerators to PyTorch/JAX and serving stacks without treating either as opaque. Whole-stack debugging: end-to-end and timeline tracing, workload replay, and reasoning from architectural constraints (SRAM ...

runs. You Might Be a Good Fit If You Have Strong engineering experience in large-scale distributed training or HPC systems. Deep familiarity with JAX internals, distributed training libraries, or custom kernels/fused ops. Experience with multi-node cluster orchestration (Slurm, Ray, Kubernetes, or similar). Comfort debugging … closely with infra, research, and deployment teams. Nice to Have Experience with training LLMs or other large transformer architectures. Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.). Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches). Experience with data pipeline optimization, sharded ...

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

Member of Technical Staff, Training Infrastructure — Inherent (London) At Inherent, we are on a mission to build AI that recursively self‐improves to discover new knowledge. Scientific advances are the backbone of our economic, technological ...

NLP PEOPLE is looking for a Machine Learning Engineer to build and deploy advanced models to optimize complex operational systems. The role involves hands-on engineering, working closely with the founding team and customer data. ...

Bjak is seeking a Technical Lead in Machine Learning to oversee the execution layer of their AI chat app. This role involves managing ML systems, ensuring they are trainable and deployable under real-world constraints. ...