26 to 50 of 110 JAX Jobs in London

Agents, reasoning models, and RAG Have high-level familiarity with the architecture and operation of large language models Have personally implemented models in PyTorch, Jax, or TensorFlow Have a strong foundation in data structures, algorithms, and software engineering principles Exhibit excellent problem-solving and analytical skills with a proactive approach ...

learning or equivalent practical experience. Experience with training generative models (Large Language Model (LLM), image, video). Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience conducting research. Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR). Preferred qualifications Experience with large-scale data pipelines. ...

learning or equivalent practical experience. Experience with training generative models (Large Language Model (LLM), image, video). Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience conducting research. Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR). Preferred qualifications: Experience with large-scale data pipelines. ...

members in diverse fields Strong programming abilities in Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL Expertise with deep learning approaches such as diffusion or flow matching Proficiency in modern software development practices: code testing, documentation, packaging ...

Legal, Compliance, and external partners like Google DeepMind.Skills and qualifications EssentialAI/ML Engineering: Deep conceptual and practical understanding of deep learning frameworks (e.g., JAX, PyTorch, TensorFlow), large-scale cloud training or inference infrastructure, and LLM ecosystems.Adversarial ML Mindset: Strong familiarity with AI security threat vectors (prompt injection, model inversion ...

Nice to have Experience with optimisation, reinforcement learning and/or large language models Familiarity with deep learning frameworks such as TensorFlow, PyTorch or JAX Experience building and maintaining research infrastructure or tools Exposure to chip design or applied AI research #J-18808-Ljbffr ...

prototypes to reliable deployment; Have experience partnering closely with researchers throughout implementation; Be proficient in Python and deep learning frameworks such as PyTorch or JAX; Have strong software engineering fundamentals, including clean, testable, well‐documented and maintainable code, version control and code reviews; Have experience with distributed training frameworks such ...

large, possibly noisy, data sets. Strong analytical and quantitative skills, and a detail-oriented mindset. Strong proficiency in machine learning libraries such as Torch, JAX or TensorFlow. Competence in Python, cluster environment, and general software engineering principles (source control, testing, collaborative workflow). Excellent written and verbal communication skills, willing ...

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play a key role ...

engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems. Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies. Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live quantitative ...

Minimum qualifications: Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting applied ...

professional software engineering experience with meaningful work on ML inference or high-performance systems.* Familiarity with at least one deep learning framework (PyTorch, JAX, TensorFlow).* Understanding of GPU architectures (memory hierarchy, warp scheduling, tensor cores).* Understanding of common LLM architectures and inference optimization techniques (e.g. quantization, speculative decoding ...

engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems. Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies. Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live quantitative ...

Switzerland . Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting applied ...

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch Distributed ...

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch Distributed ...

performance engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems.Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies.Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live quantitative strategies.Experience ...

strong programming skills in Python , Rust, or similar; and strong software engineering fundamentals building performant and reliable systems. Proficient in deep learning frameworks (Pytorch, JAX, TensorFlow). You can work on different layers of the stack from low-level training backends, data ingestion to ML/RL algorithmic design ...

engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems. Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies. Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live quantitative ...

agents, LLM research, as well as reinforcement learning and fine-tuning Strong knowledge of linear algebra, calculus and statistics.Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas.Depending on your experience: project supervision, leadership, or management.Strong knowledge of LLM landscape today ...

record of shipping products and features inside a fast-moving environment Strong coding skills in Python and familiarity with ML frameworks like PyTorch or JAX Experience training, fine-tuning, and evaluating large language models Deep familiarity with large language models and web agents, how they work, and how they ...

. Experience limited to API usage or prompt engineering is not sufficient. Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks. Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking. Evidence of research/ ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

especially knowledge in formal methods, stochastic systems and processes, the foundations of deep learning. Experience coding with deep learning libraries such as Pytorch/JAX is essential. Applicants must hold a PhD in computer science, mathematics or equivalent experience.Please see job description for a full list of requirements. What ...