76 to 100 of 124 Permanent JAX Jobs

PREFERRED QUALIFICATIONS - Knowledge of Machine Learning and LLM fundamentals, including transformer architecture, training/inference lifecycles, and optimization techniques - Knowledge of ML frameworks including JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, and TensorRT - Experience in developing and deploying LLMs in production on GPUs, Neuron, TPU or other AI acceleration hardware - Experience ...

optimize the performance of our custom hardware. Additionally, you will drive innovations that allow the profiler to support multiple frameworks, such as PyTorch, JAX, and XLA. A successful candidate will have experience in architecting, building, and operating distributed systems with a focus on high availability and fault tolerance, Hands ...

integrating ML systems into production while staying grounded in product impact and real-world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

evaluation, and testing Optimise reliability, scalability, latency, and user experience Work closely with Product and Engineering teams to solve complex customer problems Python PyTorch JAX vLLM Vector Databases What We're Looking For Strong software engineering fundamentals Experience building AI agents, copilots, RAG systems, or workflow automation platforms Excellent Python ...

high agency and low ego Strong candidates may also have Experience with large-scale distributed training (RL, pre-training, or post-training) Familiarity with JAX, PyTorch, or similar ML frameworks A track record of operating at the edge of research and infra in a fast-moving environment Compensation Annual Salary ...

Languages* Machine Learning Models* Model Training* Engineering Workflows* Data Analysis at Scale* Data Visualization* Data Ingestion Pipelines* Large Language Models (LLMs)* AI Agents* Tools* JAX* PyTorch #J-18808-Ljbffr ...

integrating ML systems into production while staying grounded in product impact and real‐world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

have strong experience building production-grade machine learning systems at scale You are experienced with modern ML frameworks such as PyTorch, TensorFlow, or JAX You have worked with or are interested in multimodal machine learning You understand how to design systems that balance automation with quality, safety, and user experience ...

models, fighting deep learning compilers). Excellent understanding of GPU architecture and why some models run faster than others. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice to have: Robotics or autonomous ...

impacting real world drug design programs.Skills and qualifications Essential:Significant experience with large scale distributed training of LLMs.Experience with deep learning ML frameworks (either JAX or PyTorch).Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL).Good understanding of GPU architectures. Reasoning about performance concepts is more important than ...

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

from data to UI. Applicants should have strong quantitative skills and experience in Python/SQL/TypeScript and PyData stack, plus PyTorch/JAX/TF for model development. #J-18808-Ljbffr ...

live training/matchday schedule, not just be technically neat. Python/SQL/TypeScript, PyData stack (Pandas/Polars, Altair, etc.), PyTorch/JAX/TF. You'll own things end-to-end, data through UI. Looking for strong quant background, first-principles engineering judgment, and taste ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

this role, you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (e.g, PyTorch, TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based servers in the Amazon cloud. You will be responsible for solving hard compiler optimization problems to achieve …/Vector engines/ML-accelerators. - Experience with OpenSource compiler toolset like LLVM/MLIR. - Experience with the following technologies: PyTorch, OpenXLA, StableHLO, JAX, TVM, deep learning models, and algorithms. - Experience with modern build systems like Bazel/CMake. Amazon is an equal opportunity employer and does not discriminate ...

this role, you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (e.g, PyTorch, TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based servers in the Amazon cloud. You will be responsible for solving hard compiler optimization problems to achieve … degree in computer science or equivalent - Proficiency with resource management, scheduling, code generation, and compute graph optimization - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains and Instruction Set Architectures Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat’spossible in asset management. Today, AI is reshaping every aspect of our industry, from ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

Olix Computing seeks an Architect/Staff Systems Software Engineer to own the DX-1 runtime and serving stack, connecting PyTorch/JAX to the hardware and enabling production inference at rack scale. This is a whole-stack role at the intersection of runtime, network, and accelerator with close collaboration ...

decisions Breadth across model architectures — transformers, diffusion, Mamba and others; the field hasn't converged and curiosity matters Fluency in Python and PyTorch or JAX Systems-level thinking: reasons from first principles, not from defaults Fully fluent in English — written and verbal, hard requirement Must be based in Europe, ideally ...