Tensor Jobs in California (NOW HIRING)

Torch, Tensor Flow Experience in Visual SLAM (Simultaneous Localization and Mapping), YOLO, Faster R CNN and deep learning models Knowledge of NVIDIA ISSAC Sim, Omniverse to develop the models Basic ...

Deep expertise in 3D parallelism (Data, Tensor, Pipeline). * Experience managing SLURM or Kubernetes-based GPU clusters. * Strong systems engineering background (C++, CUDA, Python)

Python (PyTorch/Tensor Flow) deployed on Red Hat Enterprise Linux (RHEL) tactical edge servers. * Specific GA Tech: Familiarity with DDS (Data Distribution Service) middleware (RTI Connext) used in ...

Advanced experience in a tensor/array computation library like PyTorch, TensorFlow, Jax, or similar * A detailed understanding of transformer training parallelism strategies like data parallelism ...

... Tensor Cores, and Nsight for maximum performance • Work on Linux kernel internals, scheduling, memory management, and resource isolation at cluster scale • Build custom container orchestration ...

Security is a paramount focus for Google's Tensor SoC and Pixel, especially with advancements like Gemini and Gen AI. Our team develops innovative systems security solutions for Pixel products ...

... Tensor Cores, and Nsight for maximum performance • Work on Linux kernel internals, scheduling, memory management, and resource isolation at cluster scale • Build custom container orchestration ...

Develop and optimize model-parallel training strategies (data, tensor, pipeline parallelism) with custom sharding techniques that minimize communication overhead. * Optimize GPU utilization, memory ...

Build, implement, and optimize CUDA kernels for equivariant neural network primitives - tensor products, segmented polynomials, and triangle-based operations - targeting peak performance across ...

Lead performance investigations and root-cause analysis on inference bottlenecks, e.g. memory bandwidth, cache contention, kernel launch overhead, tensor fragmentation * Establish coding patterns ...

Build, implement, and optimize CUDA kernels for equivariant neural network primitives - tensor products, segmented polynomials, and triangle-based operations - targeting peak performance across ...

... Tensor Flow, PyTorch • Experience in distributed systems specially Kafka and its eco systems like Gateways, Publisher, Consumer. • Good understanding of AWS Cloud ( S3 , RDS, IAM, Docker ...

NCCL, NVLink, InfiniBand, RDMA libraries, model/tensor parallelism. • Good if you touched any of Low-precision inference: INT8/FP8/FP4 quantization, mixed-precision serving. • Good if you touched ...

... Tensor Cores, and Nsight for maximum performance • Work on Linux kernel internals, scheduling, memory management, and resource isolation at cluster scale • Build custom container orchestration ...

Optimize endpoint configurations, KV cache behavior, batching strategy, speculative decoding, tensor parallelism, quantization, GPU utilization, latency, throughput, and cost efficiency. * Support ...

... tensor processing elements in neural accelerators). * Develop targeted exercises and benchmarks to guide AI agents in mastering environment-specific skills. * Oversee cross-functional teams ...

Graph neural networks, geometric deep learning, equivariant models, neural operators, tensor methods, manifold learning, and learning on structured state spaces. * Models that combine prediction ...

Responsibilities : • Scale distributed training across large GPU clusters (data, tensor, pipeline parallelism) • Optimize communication patterns and gradient synchronization • Improve ...

Work on distributed training systems spanning data, tensor, and pipeline parallel workloads. * Help shape the architecture of our RL training stack, including async rollout and post-training systems.

Develop configuration updates to win critical POCs, benchmarks, and optimize customer deployments; tune KV cache, apply speculative decoding, determine optimal tensor parallelism, and determine ...