Comprehensive, type-safe library for numerical computing, tabular data workflows, and machine learning. Built to be ergonomic, predictable, and strongly typed.
import { tensor, add, GradTensor } from "deepbox/ndarray"// Tensor operations with broadcastingconst a = tensor([[1, 2], [3, 4]])const b = tensor([[5, 6], [7, 8]])const c = add(a, b)// Automatic differentiationconst x = new GradTensor([2, 3], { requiresGrad: true })const y = x.mul(x).sum()y.backward()console.log(x.grad) // [4, 6]import { Sequential, Linear, ReLU } from "deepbox/nn"import { Adam } from "deepbox/optim"const model = new Sequential( new Linear(10, 64), new ReLU(), new Linear(64, 1))const optimizer = new Adam(model.parameters(), { lr: 0.001 })A complete numerical computing and ML toolkit — now with full TypeScript support and type safety.
90+ tensor operations with full broadcasting and multiple dtypes support.
GradTensor with full backward pass support for neural network training.
Tabular API with 50+ operations for data manipulation and analysis.
Linear, Conv, RNN/LSTM/GRU, Attention layers with optimizers and schedulers.
Classical ML models including Trees, SVM, KNN, Naive Bayes, and Ensembles.
40+ ML metrics, hypothesis tests, correlations, and statistical analysis.