PyTorch Lightning Bolts

PyTorchLightning/pytorch-lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.

  • pytorch
  • ai
  • machine-learning
  • image-processing
  • natural-language-processing
  • supervised-learning
  • gan

More in Python

AutoGluonawslabs/autogluon

AutoML for Image, Text, Tabular, Time-Series, and MultiModal Data.

GeneralDeep learningVisionNLPMLOps
10.8k updated 2d ago

Transformershuggingface/transformers

A deep learning library containing thousands of pre-trained models on different tasks. The goto place for anything related to Large Language Models.

Deep learningLLMsNLPSpeech
168.1k updated today

timmrwightman/pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more.

Deep learningVision
37.2k updated 4d ago

detectron2facebookresearch/detectron2

FAIR's next-generation research platform for object detection and segmentation. It is a ground-up rewrite of the previous version, Detectron, and is powered by the PyTorch deep learning framework.

Deep learningVision
34.9k updated 4d ago

Detectronfacebookresearch/Detectron Archived

FAIR's software system that implements state-of-the-art object detection algorithms, including Mask R-CNN. It is written in Python and powered by the Caffe2 deep learning framework.

Deep learningVision
26.3k archived

numpy-MLddbourgin/numpy-ml

Reference implementations of ML models written in numpy

GeneralDeep learningNLP
16.3k updated 2y ago
O

OpenFacecmusatyalab/openface

Free and open source face recognition with deep neural networks.

Deep learningVision
15.4k updated 10d ago

albumentationsalbu/albumentations Archived

А fast and framework agnostic image augmentation library that implements a diverse set of augmentation techniques. Supports classification, segmentation, detection out of the box. Was used to win a number of Deep Learning competitions at Kaggle, Topcoder and those that were a part of the CVPR workshops.

Deep learningVision
15.3k archived
All 443 in Python
Esc

Type to search every entry. ↑↓ to move, Enter to open.