Updated Official Website Read more
PyTorch library for energy-based models, diffusion, and flow matching. Implements samplers, score/flow/contrastive objectives, SDE/ODE integrators, interpolants, and mixed-precision training.
PyTorch · CUDA

High-throughput world model inference engine. Supports BiD and AR world models (DreamDojo, Cosmos, DreamZero, etc.) with persistent KV-cache, CUDA graph capture, continuous batching, speculative decoding, and torch.compile fusion. Sub-millisecond scheduling via async token queues.
PyTorch · CUDA · Python

GPU-accelerated randomized linear algebra. CUDA kernels for Hutchinson trace estimation, randomized SVD, and probabilistic matrix operations, useful for sliced score matching and large-scale ML.
C++ · CUDA · Python

Modular transformer research library. Composable attention mechanisms, positional encodings, and architecture variants for rapid prototyping.
TensorFlow

Knowledge graph embedding library. Train and evaluate TransE, DistMult, ComplEx, and other relational models on link prediction tasks.
TensorFlow 2

Bipartite graph link prediction. Implements Jaccard, Adamic-Adar, Common Neighbors, Preferential Attachment, and Katz similarity for two-mode networks.
NetworkX · NumPy