Menu
Burn WGPU backend - Cross-platform GPU acceleration using WebGPU. Supports Vulkan, Metal, Direct3D, and WebGL.
Install with Codex or Claude Copy this prompt, paste it into Codex, Claude, or another assistant, and let it review the skill page and install it for you.
Based on SOC occupation classification
Convert Markdown with LaTeX math, images, and tables into a polished PDF. Two rendering engines: (1) weasyprint — pure Python, no browser needed; (2) chromium — best for matrices/bmatrix/pmatrix/vmatrix and complex formulas. Supports Chinese/Japanese fonts. Can chain after ocr-md-polish for a complete OCR→clean→PDF workflow.
Burn core framework - Provides tensor operations, automatic differentiation, and neural network building blocks for Rust deep learning.
Burn CUDA backend - High-performance NVIDIA GPU acceleration. Provides optimal performance for NVIDIA hardware.
Burn Metal backend - Apple GPU acceleration for macOS and iOS. Optimized for Apple Silicon and Intel-based Macs.
Burn NdArray backend - CPU-based tensor operations using Rust's ndarray crate. Ideal for development, testing, and lightweight inference.
Burn ONNX support - Import ONNX models into Burn for inference on any backend. Supports compile-time code generation and runtime loading.
| name | burn-wgpu |
| description | Burn WGPU backend - Cross-platform GPU acceleration using WebGPU. Supports Vulkan, Metal, Direct3D, and WebGL. |
burn-wgpu provides cross-platform GPU acceleration using WebGPU through the wgpu crate. It supports:
use burn::tensor::Tensor;
use burn_wgpu::WgpuDevice;
// Create GPU device
let device = WgpuDevice::default();
// Create tensor on GPU
let tensor = Tensor::from_data(&[1.0, 2.0, 3.0], &device);
// GPU-accelerated operations
let result = tensor.matmul(other_tensor);
use burn_wgpu::WgpuBackend;
type Backend = WgpuBackend;
let model: Model<Backend> = Model::default();
[model_exports]
default_backend = "wgpu"
default_device = "Gpu"
[wgpu]
wgpu feature for WebAssemblyburn: Core frameworkburn-onnx: ONNX model import with GPU acceleration