← all repositories

pyg-team/pytorch-frame

A PyTorch extension for building neural networks on heterogeneous tabular data with support for diverse column types.

787 stars Python ML Frameworks
pytorch-frame
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

PyTorch Frame is a modular deep learning extension for PyTorch specialized in processing heterogeneous tabular data. It provides neural network implementations for tabular learning, supporting numerical, categorical, time, text, and image columns. The library includes benchmark datasets, mini-batch loaders, and interfaces for integrating custom data and model architectures including Large Language Models.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.