26 to 37 of 37 Google JAX Jobs

Explainable AI (XAI) SHAP/LIME interpretation, fairness flagging Data Storytelling Generating readable summaries from charts or model outputs Tools TensorFlow/PyTorch/JAX/HuggingFace We offer a pay range of $25-to- $100 per hour, with the exact rate determined after evaluating your experience, expertise ...

Explainable AI (XAI) SHAP/LIME interpretation, fairness flagging Data Storytelling Generating readable summaries from charts or model outputs Tools TensorFlow/PyTorch/JAX/HuggingFace We offer a pay range of $25-to- $100 per hour, with the exact rate determined after evaluating your experience, expertise ...

Explainable AI (XAI) SHAP/LIME interpretation, fairness flagging Data Storytelling Generating readable summaries from charts or model outputs Tools TensorFlow/PyTorch/JAX/HuggingFace We offer a pay range of $25-to- $100 per hour, with the exact rate determined after evaluating your experience, expertise ...

similar field. * Experience building, deploying and maintaining NLP/Deep Learning Systems. * Extensive experience in at least one Deep Learning framework (PyTorch, TensorFlow, JAX). * Well versed in the scientific Python ecosystem (NumPy, Scikit-Learn, Pandas etc.). * Strong Data Engineering underpinnings and an ability to work with ...

software engineering practices Demonstrated expertise in building production-quality code and data pipelines for ML systems Proficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applications Understanding of LLM training methodologies including instruction fine … generation, data curation, or curriculum learning pipelines Experience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google Cloud Hands-on experience with MLOps, experiment tracking, and model deployment systems Strong interest in staying current with ML research literature and ability ...

software engineering practices Demonstrated expertise in building production-quality code and data pipelines for ML systems Proficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applications Understanding of LLM training methodologies including instruction fine … generation, data curation, or curriculum learning pipelines Experience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google Cloud Hands-on experience with MLOps, experiment tracking, and model deployment systems Strong interest in staying current with ML research literature and ability ...

Minimum Requirements: In-depth knowledge of Python for high-performance, data-intensive applications. Proficiency with at least one modern deep learning framework (e.g., PyTorch, Jax, TensorFlow Expertise in one or more of the following areas: computer vision, large language models, recommender systems, or operations research. Foundational knowledge of statistics ...

engineers to ship fast in an environment that iterates constantly What they're looking for Strong Python engineering background Experience with frameworks like PyTorch, JAX or TensorFlow Demonstrated ability to scale ML workflows in real-world settings (cloud, GPU clusters, distributed training) Comfortable with MLOps tooling ...

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, ...) Excellent communication skills to report and present research findings and developments clearly, both orally and in writing. Curious and innovative disposition capable … writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, MS Azure, LambdaLabs 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 ...

Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...