pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Train, fine-tune, and evaluate models — the infrastructure layer.
Stack & tags
autograddeep-learninggpumachine-learningneural-networknumpypythontensor
Similar open-source projects
tensorflow
★ 200,721An Open Source Machine Learning Framework for Everyone
transformers
★ 167,239🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.