spans CUDA kernel engineering, Python library development involving both PyTorch and JAX, and direct collaboration...: Build, implement, and optimize CUDA kernels for equivariant neural network primitives — tensor products, segmented...
++, or CUDA in building scalable software systems; Deep learning frameworks including PyTorch and TensorFlow; Presenting... one or more programming languages such as Python, C, C++, or CUDA in building scalable software systems; Deep learning frameworks including...
and GPU architectures Deep expertise in CUDA programming, GPU memory hierarchies, and hardware-specific optimizations Proven... architectures, attention mechanisms, and model parallelism techniques PREFERRED EXPERIENCE: PyTorch, CUDA, TensorRT, OpenAI...
Advanced Micro Devices ⚡⚡ Sat, 23 May 2026 01:54:12 GMT
with a senior engineer leading the team's CUDA kernel and GEMM diagnostics work, owning well-scoped pieces of the codebase end..., CUDA kernels, GEMM-style workloads, CPUs, NICs or high-speed interconnects such as PCIe. Good interpersonal skills...
Senior Software Engineer — cuEquivariance
spans CUDA kernel engineering, Python library development involving both PyTorch and JAX, and direct collaboration...: Build, implement, and optimize CUDA kernels for equivariant neural network primitives — tensor products, segmented...
Nvidia ⚡ ⚡ Sun, 24 May 2026 01:09:15 GMT