ML Frameworks

A tiny deep-learning model that guesses what a file actually contains, not just what its extension claims.

Because 'the model said so' is not an explanation, SHAP uses Shapley values from game theory to assign every feature its exact contribution to any prediction.

An open standard that lets you train in PyTorch and deploy on hardware that has never heard of it.

A readable reference for how classic machine learning actually works under the hood, from backprop to genetic algorithms.

It centralizes model definitions so the same architecture works across PyTorch, JAX, vLLM, and llama.cpp without rewrites.

PyTorch exists to give researchers and engineers GPU-accelerated tensor math and automatic differentiation without forcing them to leave Python’s debugger and stack traces behind.

It teaches how LLMs work by implementing tokenization, attention, pretraining, and finetuning in pure PyTorch, one notebook at a time.

The familiar high-level API now runs on three major backends, letting you swap execution engines without rewriting model code.

Because training a transformer shouldn't require 245MB of PyTorch just to multiply matrices.

Megatron-LM splits into a reference training stack and a composable core for anyone who needs to squeeze every FLOP from a GPU cluster.

Jupyter notebooks that prove you can write a GPT with little more than high-school calculus and stubbornness.

Ray treats distributed computing as a Python primitive, then layers on libraries for training, tuning, serving, and reinforcement learning.

TRL wraps the messy zoo of post-training algorithms into clean, scalable trainers that actually run on hardware you might already own.

Companion notebooks for an 800-page quantitative finance textbook, from linear regression to deep reinforcement learning trading agents.

A volunteer-maintained Python machine learning module built on SciPy since 2007.

A PyTorch toolkit that treats robot learning like Hugging Face treats NLP: standardized datasets, pretrained policies, and one interface for many arms.

spaCy exists to move NLP models from research notebooks into production Python without rewriting the plumbing.

This library treats dense embeddings, rerankers, and sparse encoders as one retrieval stack instead of forcing a single model to do it all.

It exists to let developers run customized vision, text, and audio machine learning across mobile, web, and edge hardware without cloud round-trips.

NeMo shed its multimodal skin to focus on ASR, TTS, and speech LLMs—just as the field gets interesting.