At Tensorwave, we’re leading the charge in AI compute, building a versatile cloud platform that’s driving the next generation of AI innovation. We’re focused on creating a foundation that empowers cutting-edge advancements in intelligent computing, pushing the boundaries of what’s possible in the AI landscape.
Job Description:
TensorWave is seeking an ML Framework Engineer to lead the integration, optimization, and maintenance of PyTorch (and select AI libraries) on AMD ROCm GPUs. This role is critical in ensuring our AI cloud platform remains at the cutting edge of performance, stability, and compatibility by tracking upstream framework changes, debugging compatibility issues, and automating builds, testing, and benchmarking. You will be responsible for maintaining a registry of validated AI libraries, debugging low-level performance issues, and working with external maintainers to upstream fixes. You will collaborate with DevOps, MLOps, and AI researchers to ensure a seamless deployment and development experience across TensorWave’s infrastructure. This role is ideal for an engineer with deep PyTorch internals knowledge, strong GPU debugging experience, and a passion for optimizing AI workloads at the framework level.
We’re looking for resilient, adaptable people to join our team—folks who enjoy collaborating and tackling tough challenges. We’re all about offering real opportunities for growth, letting you dive into complex problems and make a meaningful impact through creative solutions. If you're a driven contributor, we encourage you to explore opportunities to make an impact at Tensorwave. Join us as we redefine the possibilities of intelligent computing.
What We Bring:
In addition to a competitive salary, we offer a variety of benefits to support your needs, including:
Stock Options
100% paid Medical, Dental, and Vision insurance
Life and Voluntary Supplemental Insurance
Short Term Disability Insurance
Flexible Spending Account
401(k)
Flexible PTO
Paid Holidays
Parental Leave
Mental Health Benefits through Spring Health
If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.
Contribute as a Cloud Engineer at TensorWave, providing expert support and solutions for cutting-edge AI cloud infrastructure and driving customer success.
Lead the development of AI-powered solutions at Airbnb to revolutionize global customer support as a Senior Staff Machine Learning Engineer.
Experienced Civil/Structural Engineer with a California Professional Engineering license needed at Chevron to support critical engineering projects and operations.
A temporary part-time role supporting the Center for Industrial Metal Forming through industry engagement and operational coordination at The Ohio State University.
Palo Alto Networks seeks a skilled Principal Cloud Security Engineer to drive design and automation of multi-cloud security solutions at scale.
Experienced Project Engineer needed at BRUDIS & ASSOCIATES to lead structural and bridge design projects in Maryland.
Experienced Senior Advanced Manufacturing Engineer wanted at Allegion to drive metal fabrication process improvements and manufacturing strategy across global facilities.
Support critical bridge inspections and advance your career with H&H as an Assistant Team Leader Trainee in New York.
A Field Service Engineer III role at Medtronic delivering technical expertise to support cardiac ablation systems with 75% travel in the Pacific Northwest region.
Advance aerospace technology as a Materials and Process Engineer at Northrop Grumman, applying material science expertise to high-impact defense projects.
Experienced Cyber Range Operations Engineer wanted to lead the technical execution and support of cyber event operations at the National Cyber Range Complex for Command Post Technologies.
Glacier seeks a seasoned Senior Computer Vision Engineer to enhance ML models and infrastructure for robotics driving recycling innovations.
Lead engineering projects for DOE nuclear facilities in a remote role with Sargent & Lundy’s Government Services Division.
Opportunity for an early-career HVAC Design Engineer to join a dynamic team focused on commercial and institutional mechanical system design in West Chester, PA.
Supercharge your large-scale PyTorch LLM workloads with our cloud powered by AMD MI300X
17 jobsSubscribe to Rise newsletter