5 of 5 JAX Jobs in Bristol

e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

. Hands-on experience training and deploying learned policies for complex behaviors in robots or simulated characters. Fluency with modern ML toolchains (PyTorch/JAX, RL or behavior-cloning libraries, agentic frameworks, MCP, distributed training workflows). Solid software-engineering fundamentals - tested, documented, maintainable Python & C++. Demonstrated ability to debug More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, bristol col-narrow-left Client: Founding Teams Location: bristol, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of AI over More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯