Skip to content

Instantly share code, notes, and snippets.

View lodestone-rock's full-sized avatar

Lodestone lodestone-rock

View GitHub Profile
@lodestone-rock
lodestone-rock / celeba.py
Last active August 11, 2025 04:50
AMD mi300x vs h100 probe code
import os
import wandb
from tqdm import tqdm
import argparse
import torch
import torchvision
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import DataLoader
@lodestone-rock
lodestone-rock / ramtorch_example.py
Created November 5, 2025 02:32
ramtorch single node multi gpu example
import os
import wandb
from tqdm import tqdm
import argparse
import torch
import torchvision
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import DataLoader