Upvote
Downvote
Staff Software Engineer, JAX External
Share Job
- Suggest Revision
- We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day.
- As a software engineer, you will work on a specific project critical to Google Cloud's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve.
- We do this by building libraries and tools that leverage the best of Alphabet and open-source technologies, centered around JAX.Jax is used extensively, including the largest production LLM deployments at Google.
- JAX's flexibility and parallelism APIs make it particularly powerful for large-scale distributed training/inference.
- As a founding member of the team, you'll be responsible for helping define our work and how we can most effectively empower the JAX community.
Active Job
Updated 14 days agoSimilar Job
Relevance
Active