Skip to content

Instantly share code, notes, and snippets.

View dorpxam's full-sized avatar

Max Prod dorpxam

View GitHub Profile
@dorpxam
dorpxam / gradscaler.hpp
Last active October 22, 2023 18:16
GradScaler implementation for Libtorch C++ API
#pragma once
#include <iterator>
#include <type_traits>
#include <torch/torch.h>
#include <torch/serialize/archive.h>
namespace torch {
namespace amp {