PyTorch

pytorch.org · ML Frameworks

Open source

#2 in ML Frameworks

The dominant deep-learning framework in research, with eager execution, torch.compile and a huge ecosystem.

#framework#deep-learning#python#research

Visit PyTorch ↗GitHub ↗

About the ML Frameworks category

The numerical foundations: tensor libraries, autodiff engines, compilers and inference runtimes that every model above them depends on.

PyTorch is one of 14 ml frameworks tools indexed on Axiomi. Facts on this page come from the tool's official site and public APIs; prices and features change, so confirm details on the official website before you commit.

// related

Related tools

All ML Frameworks →

Keraskeras.ioHigh-level deep-learning API that runs on JAX, TensorFlow or PyTorch backends.ML FrameworksOpen source★ 64.4k

JAXdocs.jax.devComposable transformations of NumPy-style programs: automatic differentiation, JIT compilation and parallelism on GPUs and TPUs.ML FrameworksOpen source★ 36.4k

TensorFlowtensorflow.orgGoogle's end-to-end machine learning platform, still widely deployed in production and on mobile via LiteRT.ML FrameworksOpen source★ 200.7k

tinygradtinygrad.orgDeliberately small deep-learning framework that aims to make it easy to add new accelerators.ML FrameworksOpen source★ 33.7k

MLXml-explore.github.ioApple's array framework for machine learning on Apple silicon, with unified memory and a NumPy-like API.ML FrameworksOpen source★ 28.7k

Ollamaollama.comOpen-source local model runner with a one-line pull-and-run CLI, a desktop app and a local HTTP API. Uses llama.cpp-based and MLX engines; optional paid cloud models.ML FrameworksFreemium★ 182.3k