Discover awesome ML - Deep Learning Python

bert

37,769

| updated 5 months ago | Apache License 2.0

TensorFlow code and pre-trained models for BERT

datasets

4,298

| updated about 2 months ago | Apache License 2.0

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

detectron2

25,567

| updated over 1 year ago | Apache License 2.0

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

gpt-2

21,450

| updated 7 months ago | Other

Code for the paper "Language Models are Unsupervised Multitask Learners"

keras

61,706

| updated 3 months ago | Apache License 2.0

Deep Learning for humans

maskrcnn-benchmark

9,104

| updated over 1 year ago | MIT License

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

NeMo

12,016

| updated about 2 months ago | Apache License 2.0

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

pytorch-seq2seq

1,431

| updated almost 2 years ago | Apache License 2.0

An open source framework for seq2seq models in PyTorch.

scikit-learn

59,644

| updated 3 months ago | BSD 3-Clause "New...

scikit-learn: machine learning in Python

transformers

133,705

| updated 2 months ago | Apache License 2.0

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

yolov5

45,908

| updated 10 months ago | GNU Affero Genera...

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Awesome Python ML - Deep Learning