26 to 50 of 124 Permanent JAX Jobs

Java, and their associated security considerations.Coding experience with AI & ML, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn.Familiarity with current and emerging regulations and standards, such as the EU AI Act, the Digital Services Act, the Digital Markets Act, GDPR ...

equivalent practical experience. Strong Python programming and software‐engineering skills. Experience developing machine‐learning, generative AI or LLM applications. Practical experience with PyTorch, JAX, TensorFlow or a similar framework. Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration. Knowledge of reinforcement learning, preference optimisation ...

related discipline, or equivalent practical experience.Strong Python programming and software-engineering skills.Experience developing machine-learning, generative AI or LLM applications.Practical experience with PyTorch, JAX, TensorFlow or a similar framework.Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration.Knowledge of reinforcement learning, preference optimisation or sequential decision ...

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 ...

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 ...

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 ...

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 ...

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 ...

series). 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/technical quality ...

practical experience. 5 years of experience developing, training, and deploying diffusion models and generative video architectures. Experience writing custom GPU/TPU kernels (e.g., JAX or PyTorch) for model serving. Preferred qualifications: PhD in Computer Science, Machine Learning, AI, or a related quantitative field or equivalent practical experience. Experience ...

experience, with at least 2 years in an ML or AI-adjacent role Strong Python skills and familiarity with major ML frameworks (PyTorch or JAX); comfortable reading and modifying model code Hands-on experience building and operating ML pipelines in production: data pipelines, training orchestration, evaluation, and serving Experience with ...

expertise in Pytorch. PREFERRED QUALIFICATIONS - Master's degree in computer science or equivalent - Experience in computer architecture - Previous software engineering expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training. Amazon is an equal opportunity employer and does not discriminate on the basis ...

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 ...

language models, retrieval systems and domain-specific tools. Qualifications include a degree in CS or related field, strong Python, and experience with PyTorch/JAX/TensorFlow; familiarity with agentic AI concepts and RL is essential. #J-18808-Ljbffr ...

bringExperience building ML or AI systems in productionHands-on experience with LLMs and agentic AIStrong Python, Rust, C++ or JavaExperience with PyTorch, JAX or TensorFlowExposure to LangGraph, Google ADK or similarStrong understanding of ML fundamentals and model trade-offsExperience with MLOps, multimodal AI or constrained hardware is a bonus rather ...

engineers through technical leadership, code reviews and best practices. Building systems that balance performance, latency, reliability and scalability. Technology Python PyTorch and/or JAX GPU-based model training and inference Modern machine learning infrastructure What We're Looking For Commercial experience building and deploying machine learning models into production. ...