Google JAX Jobs

1 to 25 of 47 Google JAX Jobs

AI Securtity Architect

Reading, Berkshire, South East, United Kingdom
Maclean Moore Ltd
Java, and their associated security considerations. Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle more »
Employment Type: Contract
Rate: £600
Posted:

AI Security Architect

Reading, Berkshire, United Kingdom
Maclean Moore Consulting
Java, and their associated security considerations. Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle more »
Employment Type: Contract
Rate: £500 - £600/annum
Posted:

Machine Learning Engineer

Bristol Area, South West, United Kingdom
Hybrid / WFH Options
Airbus
deploying and industrialising, scalable ML/AI solutions in production Proficiency in Docker, Kubernetes, and AWS services Experience using Deep Learning frameworks like Tensorflow, Jax, and/or Pytorch Proven experience in either NLP, Computer Vision, Anomaly Detection or Hybrid Modeling Not a 100% match? No worries! Airbus supports more »
Employment Type: Permanent, Part Time, Work From Home
Posted:

Data Scientist for Multidimensional Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Machine Learning Specialist in Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Ph.D. Student in Computational Physics and Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Computer Science Ph.D. Student for Electron Microscopy Imaging (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Machine Learning Ph.D. Student for Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Research Scientist in Electron Microscopy and Computational Imaging (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Research Associate in Electron Microscopy and Computational Imaging (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Ph.D. Student in Computational Science and Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Ph.D. Student in Computer Engineering for Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Ph.D. Student in Multidimensional Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

Ph.D. Student in Computational Imaging for Electron Microscopy (m/f/x)

Erlangen, Bayern, Germany
Friedrich-Alexander-Universität Erlangen-Nürnberg
sciences, materials science, applied mathe-matics or a related discipline. Proficiency in scientific Python programming. Experience with machine learning frameworks such as PyTorch, TensorFlow, Jax Excellent communication and interpersonal skills to be able to interact effectively with a diverse group of sci-entists and technical staff. Self-motivated and more »
Employment Type: Permanent
Salary: EUR Annual
Posted:

AI Developer

Lund, Skåne län, Sweden
IT Human Resources
related field with a strong focus on AI/ML. Extensive hands-on experience with AI/ML frameworks such as PyTorch, TensorFlow, MXNet, JAX, and proficiency in programming languages like Python. Solid understanding of deep learning architectures, neural network design, and optimization techniques for efficient model deployment. Familiarity more »
Employment Type: Any
Salary: GBP Annual
Posted:

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Seattle, 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

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

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

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

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

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

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: