Results for “pytorch-cuda-allocator”

15 skills
ssrjkk
pytorch
Builds and trains deep learning models with PyTorch, including tensors, autograd, and neural network modules.
2 · bundle
nvidia
jetson-package
Selects Jetson-compatible containers, vLLM runtime images, and Jetson AI Lab PyPI indexes based on Orin SM 8.7 vs Thor SM 11.0 and JetPack version.
2.2k · bundle
qcmuu
pytorch-lightning
High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scales from laptop to supercomputer with same code. Use when you want clean training loops with built-in best practices.
0 · bundle
k-dense-ai
pytorch-lightning
Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard, MLflow), and distributed training (DDP, FSDP, DeepSpeed) for scalable neural network training.
30.2k · bundle
orchestra-research
pytorch-lightning
Organizes PyTorch code with a Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks, and minimal boilerplate. Scales from laptop to supercomputer with the same code.
10.4k · bundle
bouclem
deep-learning
PyTorch, TensorFlow, neural networks, CNNs, transformers, and deep learning for production
7 · bundle
tianhao909
pytorch-lightning
High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scales from laptop to supercomputer with same code. Use when you want clean training loops with built-in best practices.
1 · bundle
qhjqhj00
auroc
Computes the AUROC metric using torchmetrics, handling binary, multiclass, and multilabel tasks with configurable thresholds and averaging.
3
tianhao909
pytorch-fsdp2
Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use when models exceed single-GPU memory or when you need DTensor-based sharding with DeviceMesh.
1 · bundle
orchestra-research
pytorch-fsdp2
Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use when models exceed single-GPU memory or when you need DTensor-based sharding with DeviceMesh.
10.4k · bundle
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
enuno
condor-strategy
CONDOR v2.0 — Multi-Asset Thesis Picker. Evaluates BTC, ETH, SOL, HYPE simultaneously and enters the strongest thesis. Conviction-scaled margin. DSL exit managed by plugin runtime via runtime.yaml.
1 · bundle
lovits
shot
PUA Shot — compact all-in-one PUA reference for explicit injection into sub-agents or short sessions. Strong flavor, same evidence-first behavior.
0
qcmuu
pytorch-fsdp2
Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use when models exceed single-GPU memory or when you need DTensor-based sharding with DeviceMesh.
0 · bundle
ecnu-icalk
ciou-giou
Replaces GIoU with Complete IoU (CIoU) loss in PyTorch object tracking or detection tasks, combining overlap area, center-point distance, and aspect-ratio similarity for improved bounding-box regression.
559