51 to 59 of 59 Google JAX Jobs

supervised pre training, Vision Transformers, or vector quantization methods. Hands on work with large medical or scientific imaging datasets and establishing evaluation tasks and metrics. Proficiency in PyTorch or JAX and experience running distributed training on GPUs or cloud HPC resources. Strong software engineering habits: version control, testing, continuous integration, code review, and containerization. Clear communication and ability to More ❯

ideally including transformers, diffusion models, or multimodal architectures. Experience applying ML to large multimodal biological datasets (genomics, transcriptomics, epigenomics, proteomics, single-cell data). Exceptional coding skills (Python, PyTorch, JAX, TensorFlow) and familiarity with scalable training frameworks. Experience managing large-scale distributed training across GPUs/TPUs. Background in generative AI, large language models, mechanistic interpretability, or reinforcement learning. More ❯

such as quantisation, pruning, and distillation to improve performance. Contribute to deployment, testing, and continuous model improvement. Requirements: Strong experience building and deploying ML models using PyTorch or TensorFlow; JAX is a plus. Solid background in NLP, computer vision and/or speech processing. Knowledge of model optimisation and performance tuning. Familiarity with distributed or edge AI environments. Passion More ❯

Experience across 3D graphics, rendering, or materials/textures is required Experience implementing and training diffusion or generative image models Strong Python skills and familiarity with frameworks like PyTorch, JAX, or TensorFlow Understanding of transfer learning, computer vision, and image processing techniques Ability to translate complex research into scalable, production-ready solutions Why join this team Work with a More ❯

Experience across 3D graphics, rendering, or materials/textures is required Experience implementing and training diffusion or generative image models Strong Python skills and familiarity with frameworks like PyTorch, JAX, or TensorFlow Understanding of transfer learning, computer vision, and image processing techniques Ability to translate complex research into scalable, production-ready solutions Why join this team Work with a More ❯

modern AI/ML methods (e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable AI-driven workflows and tools. More ❯

Life at UiPath The people at UiPath believe in the transformative power of automation to change how the world works. We're committed to creating category-leading enterprise software that unleashes that power. To make that happen, we need people More ❯

proficient software engineer in a least one programming language (Python or other e.g. Rust, Go, Java) • You have, hands-on experience with AI frameworks and tools (e.g., TensorFlow, PyTorch, Jax) • You have high technical engineering competence. This means being able to design complex software and make it usable in production • You have a high scientific track record in a More ❯

years of ML experience, with a strong record of transitioning research into production. Proven team leadership in a technical or research setting. Expert in Python and PyTorch; JAX is a plus. Strong foundation in data preparation, feature engineering, and model evaluation. Excellent communicator with the ability to adapt messaging for different audiences. Strategic mindset and leadership potential at the More ❯