19 of 19 Google JAX Jobs

quantitative discipline. Research experience involving machine learning, deep learning, optimisation or statistical modelling. Strong Python programming skills. Experience with frameworks such as PyTorch, TensorFlow, JAX or Keras. Experience designing experiments, evaluating model performance and interpreting results. A curious mindset and a passion for exploring emerging technologies. Excellent written ...

related frameworks• Experience on AI/ML model implementations using scikit learn, Tensor flow etc.,• Experience with Chat, IVR, Banking will be plus• Google Cloud (GCP) knowledgeAbout NTT DATANTT DATA is a $30 billion business and technology services leader, serving 75% of the Fortune Global 100. We are committed ...

Mathematics, etc.) or equivalent practical experience.ML Fundamentals: Deep understanding of machine learning principles and techniques.Framework Expertise: Strong proficiency in deep learning frameworks such as JAX or PyTorch.Modern Architectures: Hands-on experience building and working with modern model architectures (e.g., Transformers, GNNs, Diffusion Models).Full ML Lifecycle: Experience taking models ...

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

Advanced proficiency in Python and C++ within a Linux/UNIX environment. Technical Stack: Extensive practical experience with leading machine learning frameworks (e.g., PyTorch , JAX , or TensorFlow ). Research Experience: A demonstrated track record of designing and implementing complex machine learning architectures to solve rigorous, data-heavy problems. Note ...

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch ...

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

easy or obvious solutions.QualificationsStrong engineering skills, especially any of: CUDA/Triton/Pallas/CuTe DSL kernel development, lower-level PyTorch/JAX/XLA development, CUDA Graphs, FPGA/ASIC experienceMust have two or more years work experience building deep learning systems, for any domain: robotics, biology ...

ability to collaborate effectively in interdisciplinary teamsNice-to-haves:Experience in autonomous driving, robotics, or simulation systemsFamiliarity with large-scale training (e.g., FSDP, DeepSpeed, JAX)Experience with sim-to-real transfer or data-efficient learningContributions to open-source ML tools or research infrastructureWhat we offer you Attractive compensation with ...

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

orchestrationExperience deploying and operating production ML systemsComfortable working across models, infrastructure and productEnjoy working in fast-moving, early-stage environmentsTech Stack Python * PyTorch * JAX * LLMs * vLLM * Vector Databases * Modern Agent FrameworksGet in touch for more details - ncarolan@mcgregor-boyall.comMcGregor Boyall is an equal opportunity employer and do not discriminate ...

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

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

Full TimeOn-siteCambridge, UKPosted 1 week ago60,000-120,000 GBP/Year Machine Learning Engineer job in Cambridge | Salary: 60,000-120,000Hexwired has partnered with an innovative technology company developing advanced AI and ...

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

with advanced reinforcement learning topics, such as RL for sequence models, post-training, preference-based learning, or agentic systems.Familiarity with modern research stacks (e.g., JAX/Flax or PyTorch) and experience scaling experiments.Ability to be comfortable with scaling methodologies, evaluation techniques, and diagnosing complex failure modes.Ability to push projects … initiative.Strong experimental judgment, including selecting appropriate baselines and designing insightful ablations.Excellent communication skills, with a focus on clear presentation of research results.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 ...

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

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