Shypd.aiThe largest open index of AI toolspytorch_tabular is an open-source framework for deep learning models on tabular data. It provides a high-level API and leverages PyTorch Lightning for scalable training on GPU or CPU with automatic logging.
About
PyTorch Tabular offers a unified and accessible framework for applying deep learning models to tabular data. Designed with principles of low resistance usability, easy customization, and scalability, it simplifies the development and deployment of advanced models. The library integrates with PyTorch and PyTorch Lightning, enabling efficient training on both GPUs and CPUs, alongside automatic logging for experiment tracking. It supports a variety of state-of-the-art models including FeedForward Networks, NODE, TabNet, Mixture Density Networks, AutoInt, TabTransformer, GATE, GANDALF, and DANETs, as well as semi-supervised Denoising AutoEncoders. Users can also implement custom models, making it suitable for both real-world applications and research.
Capabilities
Pricing & Plans
Open Source
Free
Also listed in
