5 of 5 JAX Jobs in Wales

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

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

C++, and deep learning frameworks (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 ❯

C++, and deep learning frameworks (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 ❯

field (or equivalent practical experience). 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 integrated hardware-software systems. Preferred More ❯