NeVibeCode · 15 items
Skills and MCP tagged “pytorch”
pytorch-patterns skill This skill guides PyTorch model, training-loop, and data-loading work with practical patterns. Use it when reviewing code or tracking down runs that fail to reproduce.
Claude Code CX Codex Cursor +1
4,700installs
ctf-ai-ml skill Provides AI and machine learning attack techniques for CTF competitions, from adversarial examples to LoRA weight inspection and prompt injection.
Claude Code CX Codex Cursor +1
7,700installs
pytorch-lightning skill Guides agents through structuring PyTorch Lightning 2.6.6 code, from LightningModules to distributed trainers. Useful when scaling training across GPUs.
Claude Code CX Codex Cursor +1
2,000installs
pennylane skill Guides an agent in setting up PennyLane 0.45.1 quantum circuits, variational VQE algorithms, and hybrid neural models.
Claude Code CX Codex Cursor +1
1,900installs
pathml skill Guides computational pathology workflows using PathML 3.0.8. It handles whole-slide image tiling, tissue detection, and .h5path exports locally.
Claude Code CX Codex Cursor +1
1,900installs
diffdock skill Docks a small molecule into a protein with DiffDock and DiffDock-L, for one pair or a batch, then hands back ranked candidate poses.
Claude Code CX Codex Cursor +1
1,900installs
scvi-tools skill Guides probabilistic modelling for single-cell omics data. Use it for batch integration, cell annotation, multimodal data, and model-based uncertainty.
Claude Code CX Codex Cursor +1
1,900installs
torchdrug skill Helps build and debug TorchDrug 0.2.1 pipelines for molecular graphs and proteins. Use it when training property prediction, molecule generation, or retrosynthesis models.
Claude Code CX Codex Cursor +1
1,900installs
cellxgene-census skill A skill for programmatic access to CZ CELLxGENE Census, a vast catalog of standardized single-cell transcriptomics data. Query metadata, gene expression, and embeddings without downloading entire datasets.
Claude Code CX Codex Cursor +1
1,900installs
stable-baselines3 skill A skill for training reinforcement learning agents with Stable Baselines3. Supports PPO, SAC, DQN and other algorithms, works with Gymnasium custom environments and vectorized rollouts.
Claude Code CX Codex Cursor +1
1,900installs
torch-geometric skill Guides work with PyTorch Geometric graph neural networks, from graph data setup to classification tasks. Use it when a project relies on torch_geometric.
Claude Code CX Codex Cursor +1
1,900installs
spark-training-gotchas skill Diagnoses ten known ML training failure modes on NVIDIA DGX Spark GB10. It catches memory leaks, power throttling, and CUDA mismatches early.
Claude Code CX Codex Cursor
2,500installs
spark-environment-setup skill Sets up ML training and inference environments on NVIDIA DGX Spark with GB10, aarch64, and CUDA 13. Resolves wheel ABI mismatches and import errors.
Claude Code CX Codex Cursor
2,500installs
deepspot-m skill Reads a 224x224 H&E tile and returns predicted log1p-CPM expression for the genes you ask about. Useful when you want spatial transcriptomics from a slide you have already scanned.
Claude Code CX Codex Cursor +1
866installs
senior-computer-vision skill This skill covers computer vision pipelines for object detection, image segmentation, and model deployment. Use it when working with PyTorch, YOLO, or ONNX.
Claude Code
1,152installs