<div><b><span style="font-size:18px;">The Company</span></b></div><div>Gridmatic Inc. is a high-growth startup with offices in the Bay Area and Houston that is accelerating the clean energy transition by applying our expertise in data, machine learning, and energy to power markets. We are the rare startup that has multiple years of profitability without raising venture capital. At Gridmatic, we foster a collaborative and inclusive culture where learning and growth are constant. We move quickly, solve problems with integrity, and balance environmental responsibility with data-driven excellence.</div><div><br></div><div><span style="font-size:11pt;">We are looking for a Machine Learning Infrastructure Engineer to accelerate the decarbonization of the electricity system by building and optimizing the backbone of our ML platform. The ideal candidate will have solid expertise in machine learning, distributed systems and GPU-based training. They will design scalable, high-performance infrastructure for training, inference, and evaluation. They will push the boundaries of throughput and efficiency on large-scale time-series and weather datasets, while shaping the long-term vision of our ML platform. A successful candidate will thrive on continuous learning across engineering, ML systems, and energy markets, while contributing to a collaborative, mission-driven team.The ideal candidate must have strong deep learning fundamentals in addition to strong software engineering skills.</span></div><br><br><h3>You will:</h3><ul><li>Own a significant piece of our ML platform while rapidly building and iterating scalable, robust distributed infrastructure for ML training, inference, and evaluation on large-scale time-series and weather datasets.</li><li>Optimize throughput and cost by supporting model training and deployment across multiple clusters and clouds.</li><li>Improve the efficiency of machine learning models and other workloads by optimizing latency, throughput, and memory consumption. This involves pushing the boundaries of current hardware capabilities through techniques like GPU performance engineering.</li><li>Help define the long-term vision for Gridmatic’s ML platform.</li><li>Play a key role in mentoring junior engineers and interns, contributing to a collaborative, innovative, and growth-oriented team culture.</li></ul><h3>You must be:</h3><ul><li>A strong engineer with 3+ years of full-time industry experience working on ML systems.. You possess a deep understanding of the codebases you work in and write readable, scalable code.</li><li>Experienced in optimizing GPU throughput in deep learning models.</li><li>Experienced in distributed training and inference of large models on GPU clusters, utilizing core libraries and frameworks such as PyTorch, PyTorch Lightning, and Ray.</li><li>A self-starter with a strong sense of independence and ownership, and the capability to engineer large, robust systems from the initial design and conceptualization to productionization.</li><li>Hold a Masters or Doctorate degree in engineering or a related technical field.</li><li>A mission-driven individual who is enthusiastic about working toward a renewable grid and diving into the intersection of ML and energy. No prior energy experience required, but curiosity and a willingness to learn are must-haves!</li></ul><h3>Nice to haves:</h3><ul><li>End to end proficiency in building, maintaining, and debugging cluster infrastructure, utilizing Kubernetes and Terraform.</li><li>Expertise in identifying performance bottlenecks and designing and writing high-performance code for large-scale ML workloads.</li><li>Experience with at least one of: torch.profiler, TorchDynamo, TorchInductor, Triton, or other deep learning compiler stacks.</li><li>Understanding of GPU architectures or experience with GPU kernel programming.</li><li>Knowledge of cluster communication protocols such as nccl or gloo.</li><li>Experience working with any of the following: weather data, energy systems, time-series forecasting, electricity markets, or financial trading.</li></ul><br><div><span style="font-size:small;">#LI-DNI</span></div><div><br></div><div>Join our team and make a difference! Click below or email us at careers@gridmatic.com. </div>

<p>Gridmatic Inc. is a high-growth ClimateTech startup based in the San Francisco Bay Area that is accelerating the clean energy transition by applying our expertise in data, machine learning, and energy to power markets. We are the rare startup that has 6 years of profitable growth without raising venture capital. Gridmatic is a great place to work with a culture that values teamwork, continuous learning, diversity, and inclusion. We are environmentally and data-driven, with a growth-oriented mindset and we value integrity as much as excellence.</p>

ML Performance Engineer

<div><b><span>The Company</span></b></div><div>Gridmatic Inc. is a high-growth startup with offices in the Bay Area and Houston that is accelerating the clean energy transition by applying our expertise in data, machine learning, and energy to power markets. We are the rare startup that has multiple years of profitability without raising venture capital. At Gridmatic, we foster a collaborative and inclusive culture where learning and growth are constant. We move quickly, solve problems with integrity, and balance environmental responsibility with data-driven excellence.</div><div><br></div><div><span>We are looking for a Machine Learning Infrastructure Engineer to accelerate the decarbonization of the electricity system by building and optimizing the backbone of our ML platform. The ideal candidate will have solid expertise in machine learning, distributed systems and GPU-based training. They will design scalable, high-performance infrastructure for training, inference, and evaluation. They will push the boundaries of throughput and efficiency on large-scale time-series and weather datasets, while shaping the long-term vision of our ML platform. A successful candidate will thrive on continuous learning across engineering, ML systems, and energy markets, while contributing to a collaborative, mission-driven team.The ideal candidate must have strong deep learning fundamentals in addition to strong software engineering skills.</span></div><br><br><h3>You will:</h3><ul><li>Own a significant piece of our ML platform while rapidly building and iterating scalable, robust distributed infrastructure for ML training, inference, and evaluation on large-scale time-series and weather datasets.</li><li>Optimize throughput and cost by supporting model training and deployment across multiple clusters and clouds.</li><li>Improve the efficiency of machine learning models and other workloads by optimizing latency, throughput, and memory consumption. This involves pushing the boundaries of current hardware capabilities through techniques like GPU performance engineering.</li><li>Help define the long-term vision for Gridmatic’s ML platform.</li><li>Play a key role in mentoring junior engineers and interns, contributing to a collaborative, innovative, and growth-oriented team culture.</li></ul><h3>You must be:</h3><ul><li>A strong engineer with 3+ years of full-time industry experience working on ML systems.. You possess a deep understanding of the codebases you work in and write readable, scalable code.</li><li>Experienced in optimizing GPU throughput in deep learning models.</li><li>Experienced in distributed training and inference of large models on GPU clusters, utilizing core libraries and frameworks such as PyTorch, PyTorch Lightning, and Ray.</li><li>A self-starter with a strong sense of independence and ownership, and the capability to engineer large, robust systems from the initial design and conceptualization to productionization.</li><li>Hold a Masters or Doctorate degree in engineering or a related technical field.</li><li>A mission-driven individual who is enthusiastic about working toward a renewable grid and diving into the intersection of ML and energy. No prior energy experience required, but curiosity and a willingness to learn are must-haves!</li></ul><h3>Nice to haves:</h3><ul><li>End to end proficiency in building, maintaining, and debugging cluster infrastructure, utilizing Kubernetes and Terraform.</li><li>Expertise in identifying performance bottlenecks and designing and writing high-performance code for large-scale ML workloads.</li><li>Experience with at least one of: torch.profiler, TorchDynamo, TorchInductor, Triton, or other deep learning compiler stacks.</li><li>Understanding of GPU architectures or experience with GPU kernel programming.</li><li>Knowledge of cluster communication protocols such as nccl or gloo.</li><li>Experience working with any of the following: weather data, energy systems, time-series forecasting, electricity markets, or financial trading.</li></ul><br><div><span>#LI-DNI</span></div><div><br></div><div>Join our team and make a difference! Click below or email us at careers@gridmatic.com. </div>

ML Performance Engineer

About the Job

About the Company

Gridmatic

Similar Jobs

Data Engineer

Data Engineer

Staff ML Platform Engineer

Staff ML Platform Engineer

Data Engineer

Data Engineer

Data Engineer

Data Engineer

Senior/Staff Software Engineer, Machine Learning Infrastructure

Senior/Staff Software Engineer, Machine Learning Infrastructure

Software Engineer, Machine Learning Infrastructure

Software Engineer, Machine Learning Infrastructure

Machine Learning Engineer

Machine Learning Engineer

Senior Machine Learning Developer

Senior Machine Learning Developer

ML Ops Engineer - Data Lake & AI Infrastructure

ML Ops Engineer - Data Lake & AI Infrastructure

Machine Learning Engineer (m/f/d)

Machine Learning Engineer (m/f/d)New