Permanent Google JAX Jobs

1 to 25 of 45 Permanent Google JAX Jobs

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Fall City, 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

Mountlake Terrace, 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: