Google JAX Jobs

1 to 25 of 47 Google JAX Jobs

Applied Research Scientist, Cryo-Electron Microscopy and Tomography

London Area, United Kingdom
Hybrid / WFH Options
InstaDeep
With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Lagos, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select more »
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Woodinville, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Auburn, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Seahurst, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Snoqualmie, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Renton, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Mukilteo, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

North Bend, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Mercer Island, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Issaquah, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Kirkland, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Everett, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Pacific, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Kenmore, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Redmond, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Edmonds, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Carnation, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Hobart, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Sammamish, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Bothell, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Lynnwood, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Preston, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Milton, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Medina, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Maple Valley, Washington, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product … of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive workplace. Amazon more »
Employment Type: Permanent
Salary: USD Annual
Posted: