51 to 75 of 83 Permanent 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 ...

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

systems for data management, sharing and exploration Proficient in Python; familiarity with one or more of the following is desirable but not required: Pydantic, JAX, full-stack web development (React, TypeScript, FastAPI), cloud services and cloud-based applications; data management systems You enjoy staying up to date with current ...

delivery at scale — including workload profiling, scheduler configuration (SLURM, PBS, or equivalent), MPI/NCCL tuning, and distributed training frameworks such as PyTorch, JAX, or DeepSpeed. Deep understanding of GPU software environments: CUDA, cuDNN, NCCL, driver stacks, and the tooling required to run large‐scale AI training and inference ...

relevant discipline. Publications in top‐tier conferences (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR). Familiarity with a deep learning framework (e.g. PyTorch, JAX, TensorFlow, ...) Excellent communication skills to report and present research findings and developments clearly, both orally and in writing. Curious and innovative disposition capable … infrastructure engineering skills with supporting evidence. Experience training large‐scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud. Benefits Learning and development: On‐the‐job coaching, mentorship and learning from a world‐leading researcher as well as the opportunity to work ...

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

this role, you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (e.g, PyTorch, TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based servers in the Amazon cloud. You will be responsible for solving hard compiler optimization problems …/Vector engines/ML-accelerators. - Experience with OpenSource compiler toolset like LLVM/MLIR. - Experience with the following technologies: PyTorch, OpenXLA, StableHLO, JAX, TVM, deep learning models, and algorithms. - Experience with modern build systems like Bazel/CMake. Amazon is an equal opportunity employer and does not discriminate ...

About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat’spossible in asset management. Today, AI is reshaping every aspect of our industry, from ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

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

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

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

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefine what's possible ...

learning or equivalent practical experience. Experience with training generative models (Large Language Model (LLM), image, video). Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience conducting research. Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR). Preferred qualifications: Experience with large-scale data … pipelines. Experience with training large-scale models. Proven experience conducting research in industry About the job As an organization, Google maintains a portfolio of research projects driven by fundamental research, new product innovation, product contribution and infrastructure goals, while providing individuals and teams the freedom to emphasize specific types ...

Trainium ML accelerators. This comprehensive toolkit includes an ML compiler, runtime, and application framework that seamlessly integrates with popular ML frameworks like PyTorch and JAX enabling unparalleled ML inference and training performance. The Inference Enablement and Acceleration team is at the forefront of running a wide range of models … Experience optimizing inference performance for both latency and throughput on such large models across the stack from system level optimizations through to Pytorch or JAX is a must have. You can learn more about Neuron Key job responsibilities This role will help lead the efforts in building distributed inference ...

hackajob is partnering directly with Google to hire 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 as from collaborations with … 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 ...

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

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

AI Research Engineer (PhD) MITO AI · United Kingdom (Remote) · Open to Europe AI Research Engineer (PhD) — Generative & Multimodal AI for Creative Production Location: Remote (UK) — open to Europe, especially Madrid | Experience Level: Mid-Senior (PhD ...

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

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