Skip to main content
NeVibeCode · 15 items

Skills and MCP tagged “pytorch”

15 items
pytorch-patterns affaan-m 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 ljagiello 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 K-Dense-AI 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 Seth Hobson 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 Seth Hobson 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 K-Dense-AI 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 Alireza Rezvani 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

More in NeVibeCode

Project sponsors

Companies that keep this analytics open
Gold sponsor

Impressions, clicks and code copies — open statistics. We keep the numbers public for the same reason NeBlask itself is public. Become a sponsor