Remote MLOps Engineer for JAX/PyTorch & GPU Kernels
$35 - $45 per hourGravity Engineering Services Pvt Ltd.
Gravity Engineering Services Pvt Ltd. is seeking an MLOps Engineer to guide research and engineering teams in improving AI model performance. Candidates should have over 2 years of experience in ML infrastructure, with production knowledge of JAX or PyTorch. This is a contract position, allowing remote work with a commitment of 40 hours per week. Compensation ranges from $35 to $45 per hour, paid weekly through W-2 employment with Cincinnatus LLC. #J-18808-Ljbffr
Vacancy posted 15 hours ago
Similar jobs that could be interesting for youBased on the Remote MLOps Engineer for JAX/PyTorch & GPU Kernels in San Francisco, CA vacancy
$35 - $45 per hour
...the ground up. We're seeking talented MLOps Engineers with deep, hands‑on expertise in modern ML frameworks—specifically JAX, PyTorch, and kernel‑level programming (Pallas / Triton). This... ...writing or optimizing custom GPU kernels using Pallas (JAX) or Triton. Demonstrable...Remote jobHourly payContract workWeekday work- ...partnership with Cincinnatus LLC, is seeking a talented MLOps Engineer to join a leading AI lab's cutting-edge GenAI team. This... ...building foundational AI models and requires deep expertise in JAX, PyTorch, and kernel-level programming. Ideal candidates have 2+ years of...Remote jobFull time
- Mercor is seeking an MLOps Engineer to join their cutting-edge GenAI team. This role involves... ...tasks using modern ML frameworks like JAX and PyTorch. Applicants should have at least 2... ...infrastructure with a strong focus on kernel-level programming. Cincinnatus LLC, a...Remote job
$90 - $140 per hour
...Dorsey . Position: MLOps Engineer Expert Type:... ...hour Location: Remote Commitment: 40 hours... ...reasoning, and kernel-level optimization across... ...experience with JAX and/or PyTorch at scale. ~ Experience... ...optimizing custom GPU kernels using Pallas...Remote workContract workSummer workWeekday workMercor
New York, NY21 days ago- ...SL. in Loveland, Colorado is seeking a ML Performance Engineer to optimize machine learning stacks for speed and efficiency... ...Responsibilities include profiling workloads on multi-GPU clusters and writing custom CUDA kernels for training and inference. The ideal candidate should...SuggestedFlexible hours
Keysight Technologies SAles Spain SL.
Santa Rosa, CA3 days ago - ...technology company is looking for exceptional generalist engineers who thrive with autonomy. This fully remote role allows you to work on high-impact projects across the vLLM stack, from optimizing CUDA kernels to designing distributed orchestration systems. Ideal candidates...Remote work
Inferact
New York, NY2 days ago - ...training, from managing GPU/TPU compute and job... ...reusable and efficient JAX training pipelines. You... ...researchers and model engineers to translate ideas into... ...experience in JAX (preferred), PyTorch. Familiarity with... ...runtime optimization, custom kernels). Experience operating...
Physical Intelligence
San Francisco, CA3 days ago $60 - $100 per hour
...consulting opportunity for MLOps and ML systems... ...modern ML frameworks, JAX, PyTorch, kernel-level programming, and... ...current and upcoming remote consulting opportunities... ...development, framework-level engineering, model training... ...training efficiency, GPU utilization, and implementation...Remote jobHourly payWeekly payContract workPart timeFor contractorsFlexible hoursWeekday work$90 - $140 per hour
...Dorsey . Position: MLOps Engineer Expert Type:... ...Location: Remote Commitment:... ...systems reasoning, and kernel-level optimization across... ...production experience with JAX and/or PyTorch at scale. ~... ...optimizing custom GPU kernels using Pallas...Remote jobContract workSummer workWeekday workMercor
New York, NY21 days ago- ...GPU Kernel Engineer Our R&D team is seeking expert level GPU kernel engineers to help build the... ...based in either Gdansk or New York City. Remote work will be considered for exceptional... ...compilers or frameworks such as MLIR, Pytorch, Tensorflow, ONNX Runtime, TensorRT. (...Remote work
Makora
United States4 days ago $141k - $249k
...Examples include designing new CUDA kernels, quantization-aware training and... ...Work with researchers and ML engineers on best-practices for optimal resource... ...deep learning frameworks such as PyTorch or Jax. Skilled in profiling CPU and GPU code using tools such as PyTorch...Remote workWork at officeWork from homeFlexible hoursWaabi
United States4 days ago- Obsidian is seeking an experienced MLOps Engineer to join their cutting-edge GenAI team in Chicago. This full-time role requires deep expertise in ML frameworks, particularly JAX and PyTorch, and involves guiding research teams in enhancing AI model performance. The ideal...Full time
$191k - $253k
...Machine Learning Engineer Anduril Industries is a defense technology... ...frameworks such as PyTorch, JAX and TensorFlow ~ Experience... ...inference and vision kernels across CPU/GPU/NPU ~ Track record of developing... ...and analyzing remotely deployed software systems...Remote workFull timeWork experience placementInnovate Mississippi
United States2 days ago$161.93k - $227.33k
...Machine Learning Engineer Brisbane, California... ...in office), or remote. What You'll Do... ...such as PyTorch, TensorFlow, Jax or Scikit-learn.... ...data. Experience GPU/Accelerator programming and kernel development (such... ...Experience cultivating MLOps and ML infrastructure...Remote workWork at officeLocal area2 days per week3 days per weekFreenome
Brisbane, CA21 hours ago- Pragmatike is hiring a CUDA Kernel Engineer to develop and optimize NVIDIA CUDA kernels for a leading AI startup. This remote role focuses on maximizing GPU performance and throughput for high-scale AI systems. Candidates should have substantial experience with CUDA and...Remote jobRelocation package
- ...AI/ML infrastructure. Built by engineers, for engineers. From large-scale GPU orchestration to inference... ...GPU performance at the system and kernel level. Evaluate and compare GPU... ...modern deep learning frameworks (PyTorch, JAX, Megatron-LM, Tensort-LLM) Good...Remote work
Nebius
United States1 day ago - Pragmatike is seeking a CUDA Kernel Engineer to develop and optimize NVIDIA CUDA kernels for high-throughput AI systems. This role involves designing custom kernels, profiling GPU workloads, and resolving performance bottlenecks. Ideal candidates will have a strong understanding...Remote job
- Pragmatike is seeking a CUDA Kernel Engineer for a remote position to develop and optimize NVIDIA CUDA kernels for high-performance AI systems. The ideal candidate will have a deep understanding of GPU architecture, performance optimization strategies, and hands-on experience...Remote workRelocation package
- Pragmatike is seeking a CUDA Kernel Engineer to develop and optimize CUDA kernels for high-throughput AI systems. This remote role involves maximizing GPU performance, diagnosing PCIe bottlenecks, and working with Fortune 500 clients. Ideal candidates will have hands-on...Remote workRelocation package
$40 - $70 per hour
...Type Hybrid Department R&D - SW Kernels & Workloads Compensation CA$4... ...of AI. Job Title: Software Engineering Intern - Kernels Location:... ...Experience with ML frameworks such as PyTorch and ML packages like Numpy... ...computer architecture (CPU, GPU, custom ASICs, etc.)...Hourly payInternshipMixMode
Santa Clara, CA3 days ago- ...Senior MLOps & Data Systems Engineer As a global leader in micromobility, Lime is... ...world conditions. This is a remote position with a requirement... ...efficient utilization of GPU and compute resources.... ...experience in ML frameworks such as PyTorch or TensorFlow. ~...Remote workLocal area
Lime
United States21 hours ago $90 - $125 per hour
RLEE - Low-Level Engineering & Kernel Inference Optimization RL Environments... ...Kernel Optimization GPU/CUDA Compilers (LLVM/MLIR) PyTorch Extensions Distributed... .... This is a remote contractor role with ≥4... ...as CUTLASS, BitBLAS, or JAX/Pallas Modern C++, including...Remote workHourly payFor contractors- ...our story and Make Amazing Happen at CDW. The Senior ML / MLOps Engineer designs, builds, and operates scalable machine learning solutions... ...learning concepts, tools, and libraries such as TensorFlow, PyTorch, Scikit-learn, and MLflow. ~ Experience configuring and...Remote workLocal area
CDW
United States4 days ago $160k - $180k
...digital transformation consultancy and engineering services company that specializes in driving... ...K-$180K/Annum Job Description: MLOps - AWS+K8S, Python, MLOps, ML Infra/... ...coding skills ~ Working knowledge of PyTorch ~5+ years of ML engineering experience...Remote workApTask
United States3 days ago- ...MLOps Engineer Candidate should live within driving distance or be willing to relocate to the following areas: Wichita, KS; Lawton OK... ...Experience integrating models built with frameworks such as PyTorch, TensorFlow, or Keras into scalable pipelines. ~ Demonstrated...Remote workRelocation
Knowmadics
United States2 days ago - ...GPU Kernel Engineer Baseten powers mission-critical inference for the world's most dynamic AI companies. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge...Remote workFlexible hours
Baseten
United States21 hours ago - ...Limited is looking for a passionate ML Research Engineer in California, Maryland, to push the boundaries of GPU performance and contribute to AI-driven drug... ...development. Responsibilities include optimizing PyTorch and JAX codebases, accelerating compute with CUDA, Triton...
SLS
California, MD2 days ago - OptiTrack is seeking a Machine Learning Engineer to design, automate, and scale MLOps systems. This fully remote role involves maintaining ML training pipelines, collaborating... ..., and experience with ML frameworks like PyTorch or TensorFlow. Attractive benefits include...Remote job
$205k - $272.5k
...mining framework, is the engine that powers this... ...time inference across GPU/CPU clusters. Drive... ...and ML frameworks (PyTorch, TensorFlow, or JAX), backed by strong software... ...) and enterprise MLOps platforms.... ...this role can be fully remote. The salary range...Remote workWork at officeMotional
Las Vegas, NV7 days ago$272k - $431.25k
...We are seeking a Principal AI and ML Infra Software Engineer, GPU Clusters at NVIDIA to join our Hardware Infrastructure team... ...substantial distributed training operations using PyTorch (DDP, FSDP), NeMo, or JAX. Moreover, an in‑depth understanding of AI/ML workflows...NVIDIA Gruppe
Santa Clara, CA3 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Remote MLOps Engineer for JAX/PyTorch & GPU Kernels. Be the first to apply!
Related searches
- remote quality assurance San Francisco, CA
- remote accounts payable San Francisco, CA
- remote gis San Francisco, CA
- entry level remote San Francisco, CA
- remote medical billing part time San Francisco, CA
- sales engineer remote San Francisco, CA
- remote dba San Francisco, CA
- remote isolated San Francisco, CA
- remote hr coordinator San Francisco, CA
- remote program manager San Francisco, CA