3 days ago
San Jose, CA, USASenior
Responsibilities
- Select and optimize AI models for scalable, low-latency, real-time autonomous applications across edge and cloud deployments.
- Integrate AI frameworks into software stacks and optimize inference pipelines and model performance.
- Provide technical leadership on deep learning frameworks, GPU-accelerated computing, model compression, and deployment strategies.
- Analyze and optimize real-time AI software stack performance, including GPU-accelerated and autonomous navigation environments.
- Monitor advances in AI frameworks, foundation models, edge AI deployment, and autonomous systems integration.
Requirements
- Proven experience designing and implementing solutions with PyTorch, TensorFlow, JAX, or ONNX Runtime for autonomous applications.
- Hands-on experience with model profiling, benchmarking, quantization, pruning, and distillation.
- Deep expertise in ROCm or CUDA programming, GPU kernel optimization, and GPU memory management.
- Experience profiling end-to-end AI system performance using tools such as NVIDIA Nsight, TensorRT, or Triton Inference Server.
- Experience integrating AI models with ROS/ROS2, real-time systems, and edge AI hardware platforms such as NVIDIA Jetson.
- Strong problem-solving, debugging, collaboration, written communication, and verbal communication skills.
- Bachelor’s or Master’s degree in Electrical Engineering, Computer Engineering, Computer Science, or a closely related field.
Benefits
- Benefits are described in AMD’s benefits at a glance.
- The role is located in San Jose or Austin and is hybrid.
- The role is not eligible for visa sponsorship.
Tech Stack
PyTorchTensorFlow
Categories
ML EngineeringRobotics
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.