26 to 32 of 32 JAX Jobs

AI Scientist/Machine Learning Engineer We are looking for an AI scientist or machine learning engineer to design and implement the algorithms that power Biostream’s physiological monitoring and alert systems. You will work ...

inference, scientific workloads)Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods)Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm)Identify system bottlenecks (compute, memory, network, I/O)Build automation tools for benchmarking and reportingCollaborate with teams to align workloads with optimal … architectureRequired SkillsStrong experience with GCP (Compute Engine, GKE, Storage, Networking)Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF)Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes)Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM)Proficiency in Python, Bash ...

inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads with … Required Skills Strong experience with GCP (Compute Engine, GKE, Storage, Networking) Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF) Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes) Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM) Proficiency in Python, Bash ...

inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads with … Required Skills Strong experience with GCP (Compute Engine, GKE, Storage, Networking) Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF) Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes) Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM) Proficiency in Python, Bash ...

What You’ll Do The infrastructure powering the AI era doesn’t sell itself, and the customers building on it don’t want to be sold to. They want a peer. You’ll work as ...

NLP PEOPLE is looking for a Machine Learning Engineer to build and deploy advanced models to optimize complex operational systems. The role involves hands-on engineering, working closely with the founding team and customer data. ...

Bjak is seeking a Technical Lead in Machine Learning to oversee the execution layer of their AI chat app. This role involves managing ML systems, ensuring they are trainable and deployable under real-world constraints. ...