12 of 12 Permanent JAX Jobs

Effective problem-solver who takes initiative in complex production settings Experience with scientific computing, deep learning, big data, or health IT ontologies (e.g., PyTorch, JAX, Spark, HL7, FHIR) (desirable) Familiarity with cloud infrastructure (Azure/AWS), infrastructure as code, Kubernetes, Linux, Docker, data pipelines, and MLOps tools (desirable) Passion ...

experience. Strong expertise in deep learning, neural networks, and generative models (diffusion, GANs, transformers). Hands-on experience with modern ML frameworks (e.g., PyTorch, JAX, TensorFlow). Advanced programming skills in Python. Solid understanding of model training, evaluation, and scaling. Strong problem-solving, analytical, and communication skills. Ability to work ...

LLMs: Understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Proficiency in Python using PyTorch/TensorFlow/JAX framework. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP or Azure), an understanding of containerisation (e.g., Docker). Algorithms ...

understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud ...

comparable generative AI architectures Optional - Comprehensive technical proficiency and practical experience with leading deep learning frameworks, including advanced competency in one of PyTorch, JAX, TensorFlow, or equivalent platforms for model development and optimization ...

awareness of latency and cost tradeoffs MLOps competency — experiment tracking, model versioning, evaluation pipelines, ML CI Experience with Python and modern ML frameworks (PyTorch, JAX) Excellent verbal and written English communication skills Excellent analytical and problem-solving skills A desire to innovate and push current practice Desirable: Experience with vision ...

machine learning, statistics and/or mathematics.Strong programming skills in Python. Ideally, also proficiency in at least one major deep learning framework (e.g., PyTorch, JAX).An early track record of research (e.g., a high-quality Master's thesis, open-source contributions, or workshop papers). Excellent command of written ...

ICLR. Hands-on experience applying deep learning models to complex, ideally temporal or financial, datasets. Expertise in Python and modern ML frameworks (TensorFlow, PyTorch, JAX, etc.). Prior experience in a commercial or industry setting post-PhD, ideally in a similarly data-driven domain. Why Apply Compensation and total rewards ...

paradigms including foundation models, multi-modal systems, agent-based architectures, and RAG approaches Strong Python capability and familiarity with modern ML frameworks (PyTorch, TensorFlow, JAX) Proven experience balancing innovation (0→1) with mature product lifecycle management Strong track record operating in complex, regulated, or industrial environments Ability to translate ...

knowledge of ML evaluation, validation strategy, and failure modes including error analysis and iteration planning ● Experience with model development using PyTorch/TensorFlow/JAX ● Ability to communicate clearly across technical + scientific stakeholders ● Strong experience leading or coordinating cross-functional teams, and thinking strategically across product and science (prioritisation ...

experience strongly preferred Proven end-to-end ML experience: data pipelines, model development, deployment, and monitoring Strong Python and ML engineering skills (PyTorch or JAX) PhD in ML, Statistics, Computer Science, or related field (or equivalent research output) Ability to communicate complex ideas clearly to both technical and non-technical ...

academic background in Computer Science, AI/ML, Engineering, or related field Deep hands-on experience with Python and modern ML frameworks (PyTorch, TensorFlow, JAX) Proven track record delivering complex AI initiatives with tangible outcomes Experience working in industrial software, manufacturing, energy, IoT, or other asset-intensive environments is essential ...