Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/ethanhe42/kl-loss

Bounding Box Regression with Uncertainty for Accurate Object Detection (CVPR'19)

detection detection-algorithm detection-model detection-network object-detection pytorch

Last synced: 15 Nov 2024

https://github.com/mosaicml/streaming

A Data Streaming Library for Efficient Neural Network Training

dataset deep-learning machine-learning neural-network pytorch streaming

Last synced: 11 Nov 2024

https://github.com/yl-1993/learn-to-cluster

Learning to Cluster Faces (CVPR 2019, CVPR 2020)

cvpr2019 cvpr2020 face-clustering face-recognition gcn pytorch

Last synced: 02 Nov 2024

https://github.com/fidler-lab/polyrnn-pp-pytorch

PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)

annotation cvpr2018 deep-learning instance-segmentation labelling polygon-rnn pytorch

Last synced: 07 Aug 2024

https://github.com/stevenygd/PointFlow

PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows

3d-point-clouds computer-vision continuous-normalizing-flows machine-learning pytorch shapes

Last synced: 05 Aug 2024

https://github.com/alfredxiangwu/lightcnn

A Light CNN for Deep Face Representation with Noisy Labels, TIFS 2018

face-recognition lightcnn pytorch

Last synced: 16 Nov 2024

https://github.com/ShichenLiu/CondenseNet

CondenseNet: Light weighted CNN for mobile devices

deep-learning mobile-device pytorch

Last synced: 31 Oct 2024

https://github.com/jonasrauber/eagerpy

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code

eager-execution jax numpy python pytorch tensorflow tensorflow2

Last synced: 15 Oct 2024

https://github.com/shichenliu/condensenet

CondenseNet: Light weighted CNN for mobile devices

deep-learning mobile-device pytorch

Last synced: 13 Nov 2024

https://github.com/jayleicn/ClipBERT

[CVPR 2021 Best Student Paper Honorable Mention, Oral] Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning on image-text and video-text tasks.

cvpr2021 pytorch video-question-answering video-retrieval vision-and-language vqa

Last synced: 18 Nov 2024

https://github.com/csteinmetz1/auraloss

Collection of audio-focused loss functions in PyTorch

audio loss-functions pytorch

Last synced: 05 Aug 2024

https://github.com/HumanCompatibleAI/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 10 Nov 2024

https://github.com/ibm/ffdl

Fabric for Deep Learning (FfDL, pronounced fiddle) is a Deep Learning Platform offering TensorFlow, Caffe, PyTorch etc. as a Service on Kubernetes

ai artificial-intelligence caffe deep-learning deep-neural-networks deeplearning ibm-research-ai jupyter keras kubernetes-cluster machine-learning ml model python pytorch storage tensorflow watson

Last synced: 12 Nov 2024

https://github.com/humancompatibleai/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 01 Oct 2024

https://github.com/limbo018/DREAMPlace

Deep learning toolkit-enabled VLSI placement

deep-learning gpu-acceleration pytorch vlsi vlsi-physical-design vlsi-placement

Last synced: 15 Nov 2024

https://github.com/thunlp/openattack

An Open-Source Package for Textual Adversarial Attack.

adversarial-attacks adversarial-example natural-language-processing nlp pytorch

Last synced: 17 Nov 2024

https://github.com/wkentaro/pytorch-for-numpy-users

PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com

numpy pytorch

Last synced: 13 Nov 2024

https://github.com/NeuroDiffGym/neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

artificial-intelligence boundary-value-problem deep-learning differential-equations initial-value-problem mathematical-modelling neural-networks ode odes pde-solver physics-informed-neural-networks pinn pypi pytorch scientific-computing time-series

Last synced: 13 Nov 2024

https://github.com/antao97/dgcnn.pytorch

A PyTorch implementation of Dynamic Graph CNN for Learning on Point Clouds (DGCNN)

classification point-cloud pytorch segmentation

Last synced: 28 Oct 2024

https://github.com/open-mmlab/mmfewshot

OpenMMLab FewShot Learning Toolbox and Benchmark

few-shot-learning openmmlab pytorch

Last synced: 11 Nov 2024

https://github.com/apple/ml-aim

This repository provides the code and model checkpoints of the research paper: Scalable Pre-training of Large Autoregressive Image Models

jax large-scale-vision-models mlx pytorch

Last synced: 07 Oct 2024

https://github.com/locuslab/qpth

A fast and differentiable QP solver for PyTorch.

deep-learning machine-learning optimization pytorch quadratic-programming

Last synced: 17 Nov 2024

https://github.com/thunlp/OpenAttack

An Open-Source Package for Textual Adversarial Attack.

adversarial-attacks adversarial-example natural-language-processing nlp pytorch

Last synced: 01 Nov 2024

https://github.com/zdevito/aten

ATen: A TENsor library for C++11

pytorch tensor torch7

Last synced: 16 Nov 2024

https://github.com/iceychris/LibreASR

:speech_balloon: An On-Premises, Streaming Speech Recognition System

asr deep-learning esp32-lyrat fastai python pytorch rnn-transducer speech-recognition

Last synced: 05 Nov 2024

https://github.com/breezedeus/cnstd

CnSTD: 基于 PyTorch/MXNet 的 中文/英文 场景文字检测(Scene Text Detection)、数学公式检测(Mathematical Formula Detection, MFD)、篇章分析(Layout Analysis)的Python3 包

deep-learning math-formula-detection object-detection ocr python pytorch scene-text-detection text-detection

Last synced: 13 Nov 2024

https://github.com/lucidrains/linear-attention-transformer

Transformer based on a variant of attention that is linear complexity in respect to sequence length

artificial-intelligence attention-mechanism deep-learning pytorch transformer

Last synced: 22 Oct 2024

https://github.com/zhangxiaosong18/FreeAnchor

FreeAnchor: Learning to Match Anchors for Visual Object Detection (NeurIPS 2019)

computer-vision freeanchor neurips-2019 object-detection one-stage pytorch

Last synced: 26 Oct 2024

https://github.com/koursaros-ai/nboost

NBoost is a scalable, search-api-boosting platform for deploying transformer models to improve the relevance of search results on different platforms (i.e. Elasticsearch)

cloud deep-learning docker elasticsearch helm kubernetes machine-learning microservices nboost nlp proxy python pytorch search-api search-engine semantic-search tensorflow

Last synced: 01 Nov 2024

https://github.com/d-li14/mobilenetv2.pytorch

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

cvpr2018 deep-neural-networks imagenet mobilenetv2 pretrained-models pytorch

Last synced: 14 Nov 2024

https://github.com/nnsvs/nnsvs

Neural network-based singing voice synthesis library for research

deep-learning dnn kiritan python pytorch singing-synthesis singing-voice singing-voice-synthesis

Last synced: 05 Aug 2024

https://github.com/jaywalnut310/glow-tts

A Generative Flow for Text-to-Speech via Monotonic Alignment Search

deep-learning pytorch speech-synthesis text-to-speech tts

Last synced: 16 Nov 2024

https://github.com/volcengine/vescale

A PyTorch Native LLM Training Framework

llm-training pytorch

Last synced: 17 Nov 2024

https://github.com/Bobo-y/flexible-yolov5

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam,dcn and so on), and tensorrt

backbone cbam dcnv2 gcn hrnet moblienet neck object-detection ptq pytorch qat resnet shufflenet sparsity swin-transformer tensorrt triton-server yolov3 yolov5

Last synced: 09 Nov 2024

https://github.com/ThibaultGROUEIX/AtlasNet

This repository contains the source codes for the paper "AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation ". The network is able to synthesize a mesh (point cloud + connectivity) from a low-resolution point cloud, or from an image.

3d 3d-deep-learning computer-vision cvpr2018 geometry-processing pytorch

Last synced: 28 Oct 2024

https://github.com/lorenmt/mtan

The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].

attention-model multi-task-learning pytorch

Last synced: 15 Nov 2024

https://github.com/msurtsukov/neural-ode

Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations

jupyter-notebook neural-ode pytorch vae

Last synced: 04 Nov 2024

https://github.com/TropComplique/mtcnn-pytorch

Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks

convolutional-neural-networks face-detection pytorch

Last synced: 27 Oct 2024

https://github.com/kaituoxu/Conv-TasNet

A PyTorch implementation of Conv-TasNet described in "TasNet: Surpassing Ideal Time-Frequency Masking for Speech Separation" with Permutation Invariant Training (PIT).

audio-separation conv-tasnet permutation-invariant-training pit pytorch source-separation speech-separation tasnet

Last synced: 02 Nov 2024

https://github.com/ZhendongWang6/Uformer

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

deep-learning image-deblurring image-demoireing image-denoising image-deraining image-restoration pytorch

Last synced: 15 Oct 2024

https://github.com/rasbt/llm-workshop-2024

A 4-hour coding workshop to understand how LLMs are implemented and used

large-language-models llm pytorch

Last synced: 22 Oct 2024

https://github.com/iMoonLab/HGNN

Hypergraph Neural Networks (AAAI 2019)

complex-data deep-learning hypergraph hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/swz30/MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

attention-mechanism computer-vision eccv2020 image-denoising image-enhancement image-restoration low-level-vision multi-resolution-streams pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/iMoonLab/DeepHypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/TianhongDai/reinforcement-learning-algorithms

This repository contains most of pytorch implementation based classic deep reinforcement learning algorithms, including - DQN, DDQN, Dueling Network, DDPG, SAC, A2C, PPO, TRPO. (More algorithms are still in progress)

a2c actor-critic algorithm atari2600 ddpg deep-learning deep-reinforcement-learning dqn dueling-dqn flappy-bird ppo proximal-policy-optimization pytorch sac soft-actor-critic trpo trust-region-policy-optimization

Last synced: 07 Aug 2024

https://github.com/pytorch/torcharrow

High performance model preprocessing library on PyTorch

preprocessing python pytorch

Last synced: 07 Oct 2024

https://github.com/ymcui/MacBERT

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

bert language-model macbert nlp pytorch tensorflow transformers

Last synced: 16 Nov 2024

https://github.com/fastai/fastai2

Temporary home for fastai v2 while it's being developed

data-science deep-learning fastai jupyter machine-learning nbdev python pytorch

Last synced: 07 Aug 2024

https://github.com/wvangansbeke/lanedetection_end2end

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars

Last synced: 14 Nov 2024

https://github.com/wvangansbeke/LaneDetection_End2End

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars

Last synced: 14 Nov 2024

https://github.com/blackhc/tfpyth

Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).

machine-learning pytorch tensorflow

Last synced: 16 Nov 2024

https://github.com/chrischoy/fcgf

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor

Last synced: 14 Nov 2024

https://github.com/BlackHC/tfpyth

Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).

machine-learning pytorch tensorflow

Last synced: 28 Oct 2024

https://github.com/wanmeihuali/taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.

3d-reconstruction 3d-rendering computer-graphics computer-vision machine-learning nerf python pytorch real-time-rendering taichi

Last synced: 07 Nov 2024

https://github.com/demiseom/specaugment

A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain

data-augmentation python pytorch specaugment speech speech-recognition tensorflow

Last synced: 11 Oct 2024

https://github.com/DemisEom/SpecAugment

A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain

data-augmentation python pytorch specaugment speech speech-recognition tensorflow

Last synced: 07 Aug 2024

https://github.com/monologg/JointBERT

Pytorch implementation of JointBERT: "BERT for Joint Intent Classification and Slot Filling"

bert intent-classification joint-bert pytorch slot-filling slu transformers

Last synced: 06 Nov 2024

https://github.com/mit-han-lab/pvcnn

[NeurIPS 2019, Spotlight] Point-Voxel CNN for Efficient 3D Deep Learning

frustum-pointnet frustum-pointnets kitti point-cloud point-voxel-cnn pointnet pointnet2 pvcnn pytorch s3dis shapenet

Last synced: 28 Oct 2024

https://github.com/seungwonpark/melgan

MelGAN vocoder (compatible with NVIDIA/tacotron2)

gan neural-vocoder pytorch tts

Last synced: 30 Oct 2024

https://github.com/SamsungLabs/ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

hrnets interactive-segmentation pretrained-models pytorch segmentation

Last synced: 15 Nov 2024

https://github.com/DerrickXuNu/OpenCOOD

[ICRA 2022] An opensource framework for cooperative detection. Official implementation for OPV2V.

autonomous-driving collaborative-perception computer-vision deep-learning multi-agent-perception multi-agent-systems pytorch simulations

Last synced: 28 Oct 2024

https://github.com/lucasjinreal/dcnv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

centernet dcnv2 pytorch

Last synced: 13 Nov 2024

https://github.com/songlab-cal/tape

Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.

benchmark dataset deep-learning language-modeling protein-sequences protein-structure pytorch semi-supervised-learning

Last synced: 17 Nov 2024

https://github.com/perone/euclidesdb

A multi-model machine learning feature embedding database

cpp database deep-learning machine-learning pytorch search

Last synced: 12 Nov 2024

https://github.com/YU1ut/MixMatch-pytorch

Code for "MixMatch - A Holistic Approach to Semi-Supervised Learning"

deep-learning pytorch semi-supervised-learning

Last synced: 15 Nov 2024

https://github.com/imoonlab/deephypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 20 Oct 2024

https://github.com/JackWoo0831/Yolov7-tracker

Yolo X, v7, v8 and several Multi-Object Tracker(SORT, DeepSORT, ByteTrack, BoT-SORT, etc.) in MOT17 and VisDrone2019 Dataset. It uses a unified style and integrated tracker for easy embedding in your own projects.

pytorch

Last synced: 28 Oct 2024

https://github.com/alibaba/Megatron-LLaMA

Best practice for training LLaMA models in Megatron-LM

deepspeed distributed-training llama llm megatron-lm pretraining pytorch

Last synced: 30 Oct 2024

https://github.com/ymcui/macbert

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

bert language-model macbert nlp pytorch tensorflow transformers

Last synced: 28 Oct 2024

https://github.com/d-x-y/nas-bench-201

NAS-Bench-201 API and Instruction

automl dataset nas nas-bench-201 pytorch

Last synced: 12 Nov 2024