51 to 75 of 78 Google JAX Jobs

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

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

data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

from data to UI. Applicants should have strong quantitative skills and experience in Python/SQL/TypeScript and PyData stack, plus PyTorch/JAX/TF for model development. #J-18808-Ljbffr ...

performance, with no prior finance requirement. Candidates hold a Master’s or PhD in ML, CS, or statistics, and excel in Python, Torch/JAX/TF, #J-18808-Ljbffr ...

live training/matchday schedule, not just be technically neat. Python/SQL/TypeScript, PyData stack (Pandas/Polars, Altair, etc.), PyTorch/JAX/TF. You'll own things end-to-end, data through UI. Looking for strong quant background, first-principles engineering judgment, and taste ...

neuromorphic engineers, and keep models honest against field conditions. What you’ll bring Experience with ML for time-series or biosignals (PyTorch/JAX), model evaluation under noisy real-world data, and a bias toward deployable systems. Apply for Machine Learning Engineer, Neural Decoding — Manchester/Los Angeles #J ...

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

ensuring models stay honest against field conditions and remain deployable in real-time systems. Experience with time-series or biosignals ML using PyTorch or JAX is highly valued, with hybrid work across Manchester and Los Angeles. #J-18808-Ljbffr ...

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's possible – through powerful, trustworthy AI that ...

Switzerland . Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting … pipelines. Experience with training large-scale models. Cross-functional collaboration experience. Knowledge of machine learning, statistics, and diffusion model theories. About the job 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 ...

Minimum qualifications: Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting … pipelines. Experience with training large-scale models. Cross-functional collaboration experience. Knowledge of machine learning, statistics, and diffusion model theories. About the job 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 ...

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 as from collaborations with research programs at partner universities and technical institutes all over the world. From … active contributor to the wider research community by partnering with universities and publishing papers.Artificial intelligence will be one of humanitys 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 ...

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 as from collaborations with research programs at partner universities and technical institutes all over the world. From … contributor to the wider research community by partnering with universities and publishing papers.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 ...

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

practical experience. 5 years of experience developing, training, and deploying diffusion models and generative video architectures. Experience writing custom GPU/TPU kernels (e.g., JAX or PyTorch) for model serving. Preferred qualifications: PhD in Computer Science, Machine Learning, AI, or a related quantitative field or equivalent practical experience. Experience … inference. Proven track record of deploying machine learning models to production at scale or publishing widely adopted open-source research. About the job 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 ...

Campus AI Researcher, PhD/Postdoc (Full-Time) Full-time AI research role for PhD/postdoc candidates working on open-ended ML research, model design, tooling, and infrastructure from concept to production. For PhD ...

Join us to push the boundaries of AI and contribute to groundbreaking discoveries in a collaborative, fast-paced environment. You’ll have the opportunity to advance your career by working with the Global Technology Applied ...

Most diseases are still poorly understood at a biological level. Despite decades of research, the causal mechanisms driving many conditions remain unclear, limiting our ability to identify the right targets, design the right interventions and ...

design and scale infrastructure that powers the future of AI-driven engineering? Apply now. Responsibilities Design and scale training environments using PyTorch Distributed, JAX, or NVIDIA NeMo across multi-node/multi-GPU clusters. Manage a mixed-compute strategy spanning public clouds (AWS/Azure) and sovereign industrial clouds … that validates physical consistency laws. Skills/Must have Orchestration: Expert-level Kubernetes (AKS/EKS) is essential. ML Frameworks: Strong proficiency in PyTorch, JAX, or NVIDIA NeMo. HPC/Data: Solid experience with Python, Go, and distributed data tools (Dask/Spark). Background: Experience in AI research ...

approaches, and lessons learned You May Be a Good Fit If You Have hands-on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large-scale distributed systems Genuinely enjoy both research and engineering work—you'd describe your ideal split as roughly 50/… Care about the societal impacts of AI and responsible scaling Strong Candidates May Also Have Previous experience training LLM’s or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scale Contributed to open-source LLM frameworks (e.g., open_lm, llm-foundry, mesh-transformer-jax ...

Research Assistant or Associate in Aircraft Multidisciplinary Design Optimisation This post (Research Assistant/Associate in Aircraft Multidisciplinary Design Optimisation) will allow you to contribute to a major industry-funded project developing the next generation ...

CoreWeave is The Essential Cloud for AI. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and teams that enables innovators to build and scale AI with confidence. Trusted by leading AI ...

years of experience working in frontier AI research labs on pre-training or post-training teams. Experience writing TPU/GPU kernels (e.g., JAX, CUDA) to optimize model performance and real-time inference. Experience designing, training, and scaling generative pixel or real-time video architectures. Track record of cross … expect applicants to be experts in all fields simultaneously. 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-quality product innovation ...

experience. We do not expect applicants to be experts in all fields simultaneously. Artificial intelligence will be one of humanitys 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 … years of experience working in frontier AI research labs on pre-training or post-training teams. Experience writing TPU/GPU kernels (e.g., JAX, CUDA) to optimize model performance and real-time inference. Experience designing, training, and scaling generative pixel or real-time video architectures. Track record of cross ...