Results for “pytorch-cuda-allocator”
6 skills
k-dense-ai torch-geometric
Build and train graph neural networks with PyTorch Geometric, covering node/link/graph classification, message passing layers, heterogeneous graphs, and custom datasets.
30.2k · bundle
orchestra-research ml-training-recipes
Provides battle-tested PyTorch training recipes for LLMs, vision, diffusion, and biomedical domains, covering training loops, optimizer selection, LR scheduling, mixed precision, and debugging.
10.4k · bundle
k-dense-ai torchdrug
Build and train graph neural networks for drug discovery, protein modeling, and molecular science using PyTorch-native tools.
30.2k · bundle
k-dense-ai optimize-for-gpu
GPU-accelerate Python code using CuPy, Numba CUDA, Warp, cuDF, cuML, cuGraph, and other RAPIDS libraries for dramatic speedups on numerical, data, ML, graph, and simulation workloads.
30.2k · bundle
qhjqhj00 squad
Computes the SQuAD metric using torchmetrics, given predictions and ground truth. Use when evaluating question-answering outputs with exact match and F1 scores.
3
schattenspiegel pyarrow-python
Write, review, debug, test, or optimize Python code using PyArrow arrays, schemas, tables, compute kernels, datasets, Parquet, and Arrow IPC.
0 · bundle