Component Gallery
Browse all available components in @torchjsorg/react-ui. Each card shows a live preview of the component. Click to view full documentation or open in Storybook for interactive exploration.
Visualization
Components for visualizing tensors, attention patterns, and model outputs.
TensorViewer
Display 1D-4D tensors as heatmaps with customizable color scales
AttentionViewer
Visualize transformer attention patterns with multi-head support
EmbeddingViewer
Interactive 2D scatter plot for embedding visualization
TokenViewer
Display tokens with color-coded importance scores
HeatmapOverlay
Overlay heatmaps on images for saliency visualization
Training & Monitoring
Components for monitoring model training in real-time.
TrainingStats
Display training metrics like loss, accuracy, and progress
TrainingControls
Start, pause, and stop training with intuitive controls
LogTerminal
Display training logs with color-coded severity levels
Analysis
Components for evaluating and analyzing model performance.
ConfusionMatrix
Display classification results with optional normalization
GradientFlow
Visualize gradient magnitudes through network layers
Parameters & Input
Components for collecting user input and configuring parameters.
Charts & Plots
Components for data visualization and plotting.
Audio
Components for audio playback and recording.
Model I/O
Components for saving and loading models.
Network & Graph
Components for visualizing network architectures and graphs.
UI Components
General-purpose UI components for ML applications.
GPU-Accelerated
High-performance components that render directly from GPU memory. These require WebGPU support.
GPUHeatmap
Zero-copy heatmap rendering from WebGPU tensors
GPUPointCloud
Interactive 3D point cloud with millions of points
GPUTrainingDashboard
Multi-panel real-time training visualization
GPUStreamingChart
High-performance streaming line charts
GPUConfusionMatrix
GPU-rendered confusion matrix for large class counts
GPUAttentionFlow
GPU-accelerated attention flow visualization
GPUGradientFlow
GPU-rendered gradient flow through layers
All Components
For a complete interactive experience with all component variants and live code examples, visit our Storybook: