AI Performance Engineer

AI Performance Engineer – Remote

Bright Vision Technologies is a technology consulting and software development company delivering cloud, AI, data, and enterprise solutions across the United States.

This is a fantastic opportunity to join an established and well-respected organization offering tremendous career growth potential.

Job Title

AI Performance Engineer

Location

100% Remote (U.S.)

Position Type

Full-time, Direct W2

Salary Range

$100,000–$150,000 Annually

Experience Required

6+ years

Sponsorship

U.S. Citizens, Green Card Holders, EAD Holders, and H-1B transfer candidates are encouraged to apply. We are unable to sponsor new H-1B visa petitions for this position.

Key Responsibilities

* Profile and optimize end-to-end AI training and inference pipelines for throughput, latency, and cost. * Identify and eliminate bottlenecks across data loading, model compute, communication, and memory. * Implement and tune quantization, sparsity, and pruning strategies to reduce model footprint and accelerate inference. * Optimize distributed training using tensor parallelism, pipeline parallelism, FSDP, and ZeRO-style sharding. * Tune attention implementations using FlashAttention, paged attention, and related techniques. * Implement KV cache optimization, continuous batching, and speculative decoding for LLM serving. * Drive compiler-level optimizations using Triton, XLA, TorchInductor, or TVM, working with the broader ML framework community to land improvements that translate into measurable end-to-end performance gains. * Optimize data pipelines, sharding strategies, and storage access patterns for high-throughput training. * Build and maintain rigorous benchmark suites and regression frameworks across workloads. * Collaborate with ML and platform engineering teams to embed best practices in standard pipelines. * Drive cost-efficiency improvements through model archite