Google JAX Jobs with Hybrid or Work from Home (WFH) Options

1 to 4 of 4 Google JAX Jobs with Hybrid or WFH Options

Software Development Manager, AWS Neuron Machine Learning Distributed Training, Productivity Tooling

Cupertino, California, United States
Hybrid / WFH Options
Annapurna Labs (U.S.) Inc
is also a must. Responsible for the full development life cycle of our integrations and extensions for inference and training support in Pytorch, XLA, JAX as well as distributed training libraries like FSDP, DDP and others. Lead the way to ensure support for key ML functionality in a combined more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Machine Learning Engineer, Machine Learning and Computational Biology

London, United Kingdom
Hybrid / WFH Options
Confidential
With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, 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:

ML Engineer

London, United Kingdom
Hybrid / WFH Options
Confidential
in Ivy, and adding these to our model hub Requirements Very strong Python skills, with deep expertise in at least one of: PyTorch, TensorFlow, JAX Very strong skills in recursive programming. Check out the Ivy Container class A passion for Machine Learning research, and for our vision to unify more »
Posted:

Machine Learning Engineer in Genomics

London, United Kingdom
Hybrid / WFH Options
Confidential
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: