AI Infra2016
Unified Node.js library for web crawling and browser automation that fetches pages and files via headless browsers or raw HTTP. Provides persistent queues, proxy rotation, session management, storage, and human-like fingerprints to build scalable data pipelines (e.g., RAG/LLM datasets).
AI Deploy2016
Converts trained PyTorch, TensorFlow, and ONNX models into GPU-tuned inference engines via layer fusion, kernel auto-tuning, and reduced precision. Cuts latency, raises throughput on NVIDIA GPUs from Turing (INT8), with FP8 on Ada+ and FP4 on Blackwell+.
AI Infra2016
Provides a NumPy/SciPy-compatible GPU array library for Python, enabling existing NumPy/SciPy numerical code to run on NVIDIA CUDA and AMD ROCm with minimal changes. Exposes low-level CUDA features (RawKernels, Streams) and offers prebuilt binaries for multiple CUDA/ROCm versions.
AI Train2017
Trains gradient-boosted decision trees with native categorical-feature handling, GPU acceleration, and production-ready prediction APIs. A strong fit for tabular ML when preprocessing categories into numeric features would add noise or leakage.
AI Train2017
Sequence modeling toolkit for training custom models for translation, summarization, and language modeling. Reference implementation behind RoBERTa, BART, mBART, XLM-R, and wav2vec 2.0, with multi-GPU and mixed-precision training.
AI Image2018
Brings classic computer vision into PyTorch as differentiable, GPU-accelerated tensor operators — filters, geometric transforms, feature matching, camera calibration — so each step lives inside autograd and trains end-to-end with neural networks.
AI Model2018
Provides unified model definitions and a single API for pretrained text, vision, audio, and multimodal models for both training and inference. Emphasizes cross-framework compatibility (PyTorch/TF/JAX), pipeline-based inference, and direct access to 1M+ Hub checkpoints.
AI Model2018
Turns model definitions into a shared layer across training and inference stacks, covering text, vision, audio, video, and multimodal models. Pipelines, Trainer, and generation APIs make pretrained models usable without locking teams to one framework.
AI Coding2018
Differentiable programming framework for quantum computers: build variational circuits, compute their gradients alongside PyTorch, TensorFlow, or JAX, and run identical code on simulators or real hardware via IBM, AWS Braket, and Google plugins.
AI Infra2018
Turns NumPy-style Python into differentiable, compiled, vectorized programs for CPU, GPU, and TPU. Its leverage is composable transformations: grad, jit, vmap, and sharding combine instead of living in separate APIs.
AI Image2019
Bundles hundreds of pretrained image backbones — ResNet, EfficientNet, ViT, ConvNeXt, Swin and more — behind one consistent API for classification and feature extraction, with training and inference scripts that reproduce published ImageNet results.
AI Train2019
Turns raw PyTorch training loops into structured modules that scale from a laptop to multi-node GPUs without rewriting model logic. It handles precision, checkpointing, logging, and distributed execution while preserving PyTorch control.