remote
AI Performance Engineer - BV Teck
Software Engineer
AI Performance Engineer focused on optimizing machine learning models for speed, memory, and cost. Leverages Python, TensorFlow/PyTorch, GPU profiling, and cloud services to deliver scalable, high‑performance AI solutions.
About the role
Key Responsibilities
- Analyze and benchmark ML model performance across CPU, GPU, and distributed environments.
- Implement model optimizations such as quantization, pruning, and mixed‑precision training.
- Develop and maintain profiling pipelines using tools like NVIDIA Nsight, TensorBoard, and AWS CloudWatch.
- Collaborate with data scientists and software engineers to integrate performance improvements into CI/CD workflows.
- Document best practices and create knowledge‑sharing sessions for the engineering team.
Requirements
- Strong programming skills in Python and experience with TensorFlow or PyTorch.
- Hands‑on experience with GPU acceleration, CUDA, and performance profiling.
- Familiarity with cloud platforms (AWS, GCP, or Azure) and container orchestration (Docker, Kubernetes).
- Excellent analytical and problem‑solving abilities, with a track record of delivering measurable performance gains.
- Effective communication skills and a collaborative mindset.
Skills
pythonmachine learningtensorflowpytorchaws