Google JAX Jobs

1 to 5 of 5 Google JAX Jobs

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:

Senior Data Scientist

United Kingdom
Hybrid / WFH Options
Set2Recruit
Remote Sensing, and able to move across Domains and understand requirements, whether business or Ecology. -Proven background in Deep learning apps for Computer Vision (Jax etc) -Hands on Software Dev Skills and drive success -Passionate about Clear, clean and scaleable code If you fancy building exciting Technology and Data more »
Employment Type: Permanent, Work From Home
Salary: £85,000
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:

Compiler Engineer II - Machine Learning, Annapurna Labs

Cupertino, California, 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 … the following programming languages: C++ (preferred), C, Python PREFERRED QUALIFICATIONS - Masters or PhD degree in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, XLA/OpenXLA, MLIR Amazon is committed to a diverse and inclusive workplace. Amazon is more »
Employment Type: Permanent
Salary: USD Annual
Posted:

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: