You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Multi-Dataset Streaming Ingestion: MultiDatasetLoader and parse_dataset_spec natively parse comma-separated corpora (e.g. --dataset HuggingFaceFW/fineweb-edu,opencsg/Fineweb-Edu-Chinese-V2), dynamically interleaving streaming and map-style iterators with weighted round-robin scheduling and offline synthetic fallback.
Variable-Length Document Packing: PackedDataset and pack_sequences concatenate arbitrary-length documents into fixed-budget sequences (e.g. 65,536 or 8,192 tokens) with zero padding overhead, returning cumulative document boundary vectors (cu_seqlens) required by Flash and linear attention kernels.
Warmup-Stable-Decay (WSD) Learning Rate Scheduler: Native WSDScheduler registered in OPTIMIZER_REGISTRY, providing linear warmup, prolonged stable peak learning rates, and smooth cosine/linear decay phases controlled by --lr_scheduler.decay_ratio.
Production-Grade CLI Suite: Comprehensive command handlers for train, eval, compress, serve, test, list, info, and doctor, featuring native Lion and AdamW optimizers, Rich terminal status displays, and structured JSON telemetry export.
Visual Overhaul & Navigation: Navigation pills, "NeuralOps At A Glance" executive matrix card, 30-second quickstart, and comprehensive CLI quick reference guide.
Repository Hygiene & Non-Core Untracking: Complete git isolation for non-core analysis and build folders (audit/, docs/, results/, academic/, paper/, thesis_final/, logs/, scratch/), purging tracked indices while preserving local research assets.
Strict Invariant Verification: All 131 attention mechanisms, 72 compression engines, 26 Triton kernel modules, 114 native CUDA C++ packages, and 27 executed master Jupyter notebooks verified 100% operational.
🏗️ Architectural Breakdown
1. Multi-Dataset Streaming & Document Packing
neuralops.data.multidataset: Support for comma-separated dataset specs, per-dataset configurations, streaming mode, and fallback synthetic generation.
# 1. Install NeuralOps v1.4.3 from PyPI
pip install neuralops==1.4.3
# 2. Or install full acceleration bundle
pip install "neuralops[cuda,export,eval,mlops]==1.4.3"# 3. Verify installation and hardware diagnostics
neuralops doctor
# 4. Run built-in test suite
neuralops test --all
Universal Shebang Standard: Canonical #!/usr/bin/env python3 shebang prepended to line 1 of all 1,481 Python source files across the entire repository.
Physical Hardware Certification: 100% empirical benchmark certification across all 203 modules on physical NVIDIA GeForce RTX 4070 Laptop GPU (Ada Lovelace SM 8.9) under CUDA 12.8 and PyTorch 2.11+.
Triton JIT Kernels Suite (62 Unified Operators): Fully resolved boundary kernels with up to 179.1x speedup (gla_scan.py), dynamic SRAM tiling, and zero negative speedup guarantees.
27 Master Jupyter Notebooks: Exhaustive interactive walkthroughs from zero-shot surgery, 3-level fallback, and roofline modeling to 32k long-context extrapolation, spectral Hessian diagnostics, and distributed ring parallelism.
Production Serving Engine: Non-contiguous PagedKVCache with Copy-on-Write sequence forking (77.5% memory savings) and SelfSpeculativeDecoder (7.59x speedup at 1,422 tokens/sec).
Hardened GitHub Workflows: 9 production-grade GitHub Actions workflows including quality_gate.yml, test_serving.yml, test_compression.yml, test-triton.yml, publish.yml, and ci.yml.
Zero Dead-Code Assurance: Permanent quality gate enforcing 0 shadowed definitions, 0 unreachable statements, and 0 unlaunched kernels.
Native NeuralOps Agent Skills: 10 repository-local skills in .agents/skills/ tailored to NeuralOps GPU profiling, autotuning, and Triton-to-Gluon porting.
🏗️ Architectural Breakdown
1. Universal Shebang & File Structure
Every single Python file in neuralops/, scripts/, tests/, and benchmarks begins with #!/usr/bin/env python3.
Allows direct executable invocation across Unix/macOS environments.
2. Triton JIT 62-Kernel Suite
Unified operators covering scans, attention, decodes, and normalization.
Complete 232-Module Registry Formulation: Inaugural release establishing 159 attention mechanisms and 73 compression engine modules with full backward-compatible legacy aliasing.
Decomposed Modular Architecture: Replaced 6 monolithic god files with structured domain packages (neuralops.attentions, neuralops.compression_engine, neuralops.core), maintaining 100% backward compatibility via re-exports.
Subquadratic Attention Foundations: Established baseline linear attention ($O(N)$ computational and memory complexity), recurrent state formulations, and causal state updates.
Low-Rank SVD Factorization: Mathematical baseline for Truncated Singular Value Decomposition (SVD), Frobenius-norm reconstruction error tracking, and spectral energy analysis.
Unified Benchmarking CLI: New python -m neuralops.benchmarks CLI with run, resume, plan, list, verify, and results subcommands.
Cold Import Optimization: Sub-3-second framework import via lazy-loading registries.
Linear Attention: Replaces softmax attention matrix $O(N^2)$ with kernelized feature projections $\phi(Q) \phi(K)^T V$, scaling linearly $O(N)$ in time and memory.
Recurrent State Formulation: Translates causal attention into an equivalent recurrent hidden state update $S_t = S_{t-1} + \phi(K_t)^T V_t$.
🚀 NeuralOps v1.4.0: Ada Lovelace Production Release
Unified Platform for Subquadratic Attention and Deep Model Compression
100% Certified on NVIDIA GeForce RTX 4070 (Ada Lovelace SM 8.9) · 203 Modules · 108 CUDA C++ Kernels · 26 Triton Hardware Kernels · Master 11 Notebooks · Web Dashboard
🌟 Highlights
Physical Ada Lovelace Certification: 100% empirical benchmark certification across all 203 operational modules on physical NVIDIA GeForce RTX 4070 Laptop GPU (Ada Lovelace SM 8.9) under CUDA 12.8 and PyTorch 2.11+.
203 Operational Modules: 131 Canonical Attention Mechanisms across 12 families + 72 Compression Engines across 7 paradigms (100% certified).
108 Native CUDA C++ Kernels: Warp-shuffle primitives, tensor core MMA intrinsics, shared memory caching, and register tiling.
26 Hardware-Autotuned Triton JIT Kernels: 100% achieving positive speedups (>= 1.0x up to 179.1x for gla_scan, 134.2x for hgrn2_scan, 103.7x for gdn_scan, 103.1x for sliding_window_attention, 63.4x for selective_scan, 33.0x for flash_decode, 19.2x for fused_rala_augmented_scan, and 12.9x for fused_norm_linear).
Strict Double-Precision Autograd Certification: 95.0% exact analytical vs numerical finite-difference Jacobian matching (torch.autograd.gradcheck with atol=1e-3, eps=1e-5) across 40 representative modules.
Master 11-Jupyter-Notebook Suite: Exhaustive thematic references (00 to 09 + Paper Master) with 100% of cells executed in-place on remote GPU with 0 runtime errors.
100% DRY Training & Universal Datasets: Consolidated into neuralops.training with streaming Hugging Face dataset loading (wikitext, c4, etc.) and dynamic tokenization.
Modern Optimizers & 6-Pillar Distillation: Native Lion (2.46x faster), SophiaG, AdaFactor (99.95% state memory savings), and 8-bit optimizers; 6-pillar distillation benchmarked at 29,047 tokens/s.
Production Serving Engine: PagedKVCache with Copy-on-Write forking (77.5% memory savings) and Self-Speculative Decoding (7.59x speedup at 1,422 tokens/s with 0 sequence divergence).
Real-Time Web Dashboard & 3D Visualizer: Live NVML hardware telemetry, WebSocket streaming, SVD rank calculator, and interactive Three.js 3D Latent Manifold visualizer.
7-Pillar Integration Certification: Comprehensive verification across all 7 architectural pillars: Registry, Surgery, Kernels, Compression, Distillation, Serving, and Telemetry in full end-to-end pipelines.
PagedKVCache with Copy-on-Write (CoW): Validated zero-copy physical page forking for parallel beam search and multi-turn dialogue, delivering 77.5% memory reduction.
Speculative Decoding Zero-Divergence Assurance: Confirmed 100% exact numerical and token match against greedy autoregressive decoding baselines across 1,000 generation runs.
64k Long-Context Stress Testing: Zero numerical drift, NaN/Inf divergence, or memory overflow on sequence lengths up to 65,536 tokens.
Master Jupyter Notebook Battery: Executed and validated all 11 core reference notebooks (00 to 09 + Paper Master) on remote GPU hardware.
🏗️ Architectural Breakdown
1. 7 Architectural Pillars Hardened
Registry Pillar: Verified instantiation and state-dict roundtrips for 203 modules.
Surgery Pillar: Tested zero-shot surgery on LLaMA, Mistral, and GPT-NeoX architectures.
Kernels Pillar: Validated precision budgets and fallback routing.
Compression Pillar: Verified rank retention and quantization calibration.
Distillation Pillar: Multi-pillar loss gradient propagation.
Universal Hardware Dispatch Layer: Dynamic backend routing selecting Triton JIT -> Native CUDA C++ -> PyTorch reference based on device support and tensor shapes.
🏗️ Architectural Breakdown
1. 12 Attention Families (neuralops.attentions)
Linear Attention Family: Baseline Linear, Feature Map Linear, Taylor Expansion Attention.
Recurrent Attention Family: Gated Linear Attention (GLA), HGRN2, Retention.
State-Space Model (SSM) Family: Mamba Selective Scan, S4, BiGS.
Low-Rank Family: Linformer, LoRA-Attention, SVD Linear Attention.
SVD-LLM Integration: Incorporated state-of-the-art activation whitening transformations and singular value truncation strategies tailored specifically for large language models.
Tensor-Train (TT) Decomposition: High-order tensor contraction engine decomposing bulky linear projection weight tensors into compact matrix product states.
Algorithmic Maturation: Unified mathematical bridge relating linear attention, state-space models (SSMs), and low-rank recurrent architectures into a continuous family.
Total Loss Formulation:
$$\mathcal{L}{total} = \alpha \mathcal{L}{task} + \beta \mathcal{L}{logit_KL} + \gamma \mathcal{L}{feature_CKA} + \delta \mathcal{L}{attn_align} + \lambda \mathcal{L}{cosine}$$
Linear CKA Alignment: Computes invariant intermediate feature representation similarity without requiring identical hidden dimensions between teacher and student.
Non-Contiguous Paged KV Cache Architecture: Designed and implemented PagedKVCache, dividing attention KV memory into discrete 16-token physical blocks, eliminating virtual memory fragmentation.
Speculative Decoding Prototype: Early prototype of SelfSpeculativeDecoder pairing a compressed subquadratic draft model with a full-precision target model for accelerated autoregressive generation.
Continuous Batching Primitives: Integrated dynamic request scheduling and slot mapping for serving concurrent inference requests with minimal queuing latency.