Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-07-29 13:55:27 UTC
- JSON Representation
https://github.com/Tencent/PatrickStar
PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.
bert gpt nlp pretrained-models pytorch
Last synced: 03 Aug 2024
https://github.com/soapisnotfat/super-resolution
collection of super-resolution models & algorithms
fsrcnn pytorch resolution-models srcnn srgan sup-pixel-cnn
Last synced: 05 Aug 2024
https://github.com/Spijkervet/SimCLR
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
contrastive-learning pytorch representation-learning simclr unsupervised-learning
Last synced: 01 Aug 2024
https://github.com/loicland/superpoint_graph
Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs
clustering large-scale lidar partition ply-files point-cloud pytorch segmentation semantic semantic-segmentation superpoint-graphs
Last synced: 03 Aug 2024
https://github.com/sail-sg/Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
adan artificial-intelligence bert-model convnext cuda-programming deep-learning diffusion dreamfusion fairseq gpt2 llm-training llms mae moe optimizer pytorch resnet timm transformer-xl vit
Last synced: 01 Aug 2024
https://github.com/guochengqian/PointNeXt
[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies
3d-vision pointcloud pointnet2 pytorch
Last synced: 31 Jul 2024
https://github.com/braindotai/Watermark-Removal-Pytorch
🔥 CNN for Watermark Removal using Deep Image Prior with Pytorch 🔥.
artefacts-removal deep-image-priors deep-learning paper pytorch restoration-tasks skip-connections watermark watermark-removal
Last synced: 01 Aug 2024
https://github.com/RahulSChand/gpu_poor
Calculate token/s & GPU memory requirement for any LLM. Supports llama.cpp/ggml/bnb/QLoRA quantization
ggml gpu huggingface language-model llama llama2 llamacpp llm pytorch quantization
Last synced: 01 Aug 2024
https://github.com/williamFalcon/test-tube
Python library to easily log experiments and parallelize hyperparameter search for neural networks
caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow
Last synced: 31 Jul 2024
https://github.com/jackroos/VL-BERT
Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".
bert iclr2020 pre-training pytorch representation-learning self-supervised-learning vision-and-language vl-bert
Last synced: 01 Aug 2024
https://github.com/williamfalcon/test-tube
Python library to easily log experiments and parallelize hyperparameter search for neural networks
caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow
Last synced: 25 Sep 2024
https://github.com/kymatio/kymatio
Wavelet scattering transforms in Python with GPU acceleration
deep-learning keras numpy pytorch scattering-transform scikit-learn tensorflow wavelets
Last synced: 26 Sep 2024
https://github.com/jacobgil/vit-explain
Explainability for Vision Transformers
deep-learning explainable-ai pytorch transformer vision-transformer
Last synced: 01 Aug 2024
https://github.com/mosecorg/mosec
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
cv deep-learning gpu hacktoberfest jax llm llm-serving machine-learning machine-learning-platform mlops model-serving mxnet nerual-network python pytorch rust tensorflow tts
Last synced: 01 Aug 2024
https://github.com/msnh2012/Msnhnet
🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.
darknet inference-engine jetson-nx mobilenetv2 mobilenetyolo pytorch yolov3 yolov4 yolov5
Last synced: 02 Aug 2024
https://github.com/tusen-ai/SST
Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).
3d-object-detection autonomous-driving pytorch
Last synced: 31 Jul 2024
https://github.com/ikergarcia1996/Self-Driving-Car-in-Video-Games
A deep neural network that learns to drive in video games
autonomous-driving deep-learning deep-neural-network machine-learning neural-network pretrained-models python pytorch self-driving-car supervised-learning video-games videogame videogame-bot
Last synced: 24 Aug 2024
https://github.com/quic/sense
Enhance your application with the ability to see and interact with humans using any RGB camera.
activity-recognition calorie-estimation computer-vision deep-learning fitness-app gesture-recognition neural-networks pytorch video
Last synced: 05 Aug 2024
https://github.com/DrSleep/light-weight-refinenet
Light-Weight RefineNet for Real-Time Semantic Segmentation
deep-learning light-weight pytorch refinenet semantic-segmentation
Last synced: 04 Aug 2024
https://github.com/ritchieng/deep-learning-wizard
Open source guides/codes for mastering deep learning to deploying deep learning in production in PyTorch, Python, Apptainer, and more.
apptainer deep-learning llamaindex ollama python3 pytorch
Last synced: 26 Sep 2024
https://github.com/MisaOgura/flashtorch
Visualization toolkit for neural networks in PyTorch! Demo -->
cnn deep-learning explainability interpretability machine-learning neural-networks pytorch visualization
Last synced: 31 Jul 2024
https://github.com/PaddlePaddle/X2Paddle
Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)
caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo
Last synced: 01 Aug 2024
https://github.com/chengtan9907/openstl
OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning
artificial-intelligence attention-mechanism awesome-list awesome-lists benchmark computer-vision deep-learning mlp predictive-learning pytorch self-supervised-learning transformer video-prediction weather-forecast
Last synced: 25 Sep 2024
https://github.com/Denis2054/Transformers-for-NLP-2nd-Edition
Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more
bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax
Last synced: 29 Jul 2024
https://github.com/fboulnois/stable-diffusion-docker
Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.
dall-e dalle diffusion docker generative-art huggingface image-generation inpainting midjourney pytorch stable-diffusion tensorflow text-to-image
Last synced: 01 Aug 2024
https://github.com/nlp-uoregon/trankit
Trankit is a Light-Weight Transformer-based Python Toolkit for Multilingual Natural Language Processing
adapters artificial-intelligence deeplearning dependency-parsing language-model lemmatization machine-learning morphological-tagging multilingual natural-language-processing nlp part-of-speech-tagging pytorch sentence-segmentation tokenization universal-dependencies xlm-roberta
Last synced: 31 Jul 2024
https://github.com/creafz/pytorch-cnn-finetune
Fine-tune pretrained Convolutional Neural Networks with PyTorch
convolutional-neural-networks deep-learning pretrained-models pytorch
Last synced: 03 Aug 2024
https://github.com/primeqa/primeqa
The prime repository for state-of-the-art Multilingual Question Answering research and development.
ai bert dpr ibm ibm-research-ai language-model machine-learning natural-language-processing neural-information-retrieval neural-search nlp python pytorch question-answering semantic-search squad transfer-learning
Last synced: 01 Aug 2024
https://github.com/cloneofsimo/minDiffusion
Self-contained, minimalistic implementation of diffusion models with Pytorch.
Last synced: 31 Jul 2024
https://github.com/linksense/LightNet
LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)
cityscapes deeplabv3 deeplabv3plus inplace-activated-batchnorm light-weight-net light-weight-networks mapillary-vistas-dataset mobilenet mobilenet-v2 mobilenetv2 mobilenetv2plus pytorch rfmobilenetv2plus scse-aspp segmentation semantic-context-loss semantic-segmentation shufflenet wide-residual-networks
Last synced: 05 Aug 2024
https://github.com/zhunzhong07/Random-Erasing
Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST
aaai2020 data-augmentation image-classification object-detection person-re-identification pytorch
Last synced: 02 Aug 2024
https://github.com/microsoft/CameraTraps
PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.
camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife
Last synced: 08 Aug 2024
https://github.com/microsoft/tutel
Tutel MoE: An Optimized Mixture-of-Experts Implementation
mixture-of-experts moe nlp pytorch transformer
Last synced: 25 Sep 2024
https://github.com/benedekrozemberczki/SimGNN
A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
attention-mechanism deep-learning gcn ged gnn graph-attention graph-classification graph-convolution graph-edit-distance graph-embedding graph-similarity machine-learning network-embedding neural-network pytorch simgnn sklearn tensor-network tensorflow wsdm
Last synced: 08 Aug 2024
https://github.com/run-house/runhouse
The fastest way to iterate and deploy AI workloads on your own infra. Unobtrusive, debuggable, PyTorch-like APIs.
api artificial-intelligence aws azure collaboration data-science deployment distributed fastapi gcp infrastructure machine-learning middleware observability python pytorch ray sagemaker serverless
Last synced: 24 Sep 2024
https://github.com/onepanelio/onepanel
The open source, end-to-end computer vision platform. Label, build, train, tune, deploy and automate in a unified platform that runs on any cloud and on-premises.
ai aiops annotation computer-vision deeplearning etl hyperparameter-tuning inference jupyterlab labeling machinelearning mlops pipelines pytorch tensorboard tensorflow training workflows
Last synced: 01 Aug 2024
https://github.com/SMILELab-FL/FedLab
A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.
deep-learning federated-learning federated-learning-framework fedlab machine-learning pytorch pytorch-federated-learning
Last synced: 02 Aug 2024
https://github.com/zhiqwang/yolort
yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.
deployment detection graghsurgeon inference jit libtorch lightning ncnn nms onnx onnxruntime pytorch tensorrt torchscript trt tvm visualization yolo yolort yolov5
Last synced: 31 Jul 2024
https://github.com/lonePatient/albert_pytorch
A Lite Bert For Self-Supervised Learning Language Representations
albert bert language-model mask ngram nlp pytorch
Last synced: 01 Aug 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: 01 Aug 2024
https://github.com/cresset-template/cresset
Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.
build cuda deep-learning deep-learning-tutorial docker docker-compose machine-learning makefile mlops mlops-template python pytorch source source-python template template-repository wheel
Last synced: 01 Aug 2024
https://github.com/jia-zhuang/pytorch-multi-gpu-training
整理 pytorch 单机多 GPU 训练方法与原理
Last synced: 01 Aug 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/yl-1993/learn-to-cluster
Learning to Cluster Faces (CVPR 2019, CVPR 2020)
cvpr2019 cvpr2020 face-clustering face-recognition gcn pytorch
Last synced: 01 Aug 2024
https://github.com/AstraZeneca/chemicalx
A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)
biology chemistry deep-chemistry deep-learning drug drug-discovery drug-interaction drug-pair geometric-deep-learning geometry graph-neural-network machine-learning pharma polypharmacy pytorch smiles smiles-strings torch torchdrug
Last synced: 31 Jul 2024
https://github.com/Cadene/vqa.pytorch
Visual Question Answering in Pytorch
clevr coco deep-learning pytorch resnet skipthoughts torch vgenome vqa
Last synced: 01 Aug 2024
https://github.com/wohlert/semi-supervised-pytorch
Implementations of various VAE-based semi-supervised and generative models in PyTorch
generative-models pytorch semi-supervised-learning
Last synced: 03 Aug 2024
https://github.com/ShichenLiu/CondenseNet
CondenseNet: Light weighted CNN for mobile devices
deep-learning mobile-device pytorch
Last synced: 31 Jul 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: 25 Sep 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: 03 Aug 2024
https://github.com/csteinmetz1/auraloss
Collection of audio-focused loss functions in PyTorch
Last synced: 05 Aug 2024
https://github.com/weiaicunzai/Bag_of_Tricks_for_Image_Classification_with_Convolutional_Neural_Networks
experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc
Last synced: 01 Aug 2024
https://github.com/mhamilton723/STEGO
Unsupervised Semantic Segmentation by Distilling Feature Correspondences
computer-vision deep-learning iclr2022 pytorch semantic-segmentation unsupervised-learning
Last synced: 03 Aug 2024
https://github.com/braindecode/braindecode
Deep learning software to decode EEG, ECG or MEG signals
deep-learning ecog eeg electrocorticography electroencephalogram electroencephalography magnetoencephalography meg neuroimaging neuroscience python pytorch
Last synced: 02 Aug 2024
https://github.com/wkentaro/pytorch-for-numpy-users
PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com
Last synced: 25 Sep 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: 31 Jul 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: 01 Aug 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: 25 Sep 2024
https://github.com/qqlu/Entity
EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation
cnn computer-vision condinst deep-learning detectron2 fcos image-segmentation instance-segmentation object-detection panoptic-segmentation pretrained-models pretrained-weights pretraining pytorch segmentation semantic-segmentation
Last synced: 31 Jul 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 Aug 2024
https://github.com/Santosh-Gupta/SpeedTorch
Library for faster pinned CPU <-> GPU transfer in Pytorch
cpu-gpu-transfer cpu-pinned-tensors cuda cuda-tensors cuda-variables cupy data-transfer embeddings embeddings-trained gpu gpu-transfer machine-learning natural-language-processing nlp pinned-cpu-tensors pytorch pytorch-tensors pytorch-variables sparse sparse-modeling
Last synced: 03 Aug 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/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: 30 Jul 2024
https://github.com/open-mmlab/mmfewshot
OpenMMLab FewShot Learning Toolbox and Benchmark
few-shot-learning openmmlab pytorch
Last synced: 03 Aug 2024
https://github.com/zjunlp/PromptKG
PromptKG Family: a Gallery of Prompt Learning & KG-related research works, toolkits, and paper-list.
awsome-list demo-tuning dialogue genkgc knowledge-graph knowledge-informed-prompt-learning lambdakg link-prediction natural-language-processing nlp paper paper-list prompt-tuning promptkg pytorch question-answering relation-extraction retrieval-augmented retrievalre retroprompt
Last synced: 03 Aug 2024
https://github.com/cv-rits/MonoScene
[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image
2d-to-3d computer-vision cvpr2022 cvpr22 deep-learning kitti-360 mayavi monocular nyu-depth-v2 occupancy-prediction pytorch semantic-kitti semantic-scene-completion semantic-scene-understanding single-image-reconstruction
Last synced: 10 Aug 2024
https://github.com/MaximeVandegar/Papers-in-100-Lines-of-Code
Implementation of papers in 100 lines of code.
3d aes artificial-intelligence deep-learning diffusion-models educational gans generative-model implementation-of-research-paper inverse-rendering machine-learning meta-learning nerf neural-radiance-fields papers python pytorch reinforcement-learning research rl
Last synced: 04 Aug 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 Aug 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: 02 Aug 2024
https://github.com/modal-labs/modal-examples
Examples of programs built using Modal
cloud distributed gpu machine-learning modal python pytorch serverless stable-diffusion web
Last synced: 01 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: 02 Aug 2024
https://github.com/astra-vision/MonoScene
[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image
2d-to-3d computer-vision cvpr2022 cvpr22 deep-learning kitti-360 mayavi monocular nyu-depth-v2 occupancy-prediction pytorch semantic-kitti semantic-scene-completion semantic-scene-understanding single-image-reconstruction
Last synced: 31 Jul 2024
https://github.com/shadow2496/VITON-HD
Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)
Last synced: 01 Aug 2024
https://github.com/Mariewelt/OpenChem
OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research
computational-biology computational-chemistry deep-learning deep-neural-networks drug-design drug-discovery graph-convolutional-networks machine-learning predictive-modeling pytorch qsar
Last synced: 02 Aug 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: 03 Aug 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: 02 Aug 2024
https://github.com/locuslab/qpth
A fast and differentiable QP solver for PyTorch.
deep-learning machine-learning optimization pytorch quadratic-programming
Last synced: 01 Aug 2024
https://github.com/hollobit/All-About-the-GAN
All About the GANs(Generative Adversarial Networks) - Summarized lists for GAN
adversarial-nets adversarial-networks arxiv cgan classification ct-denoising deep-learning detection gan generative-adversarial-network image-synthesis medical-image-synthesis medical-imaging neural-network paper pytorch reconstruction segmentation tensorflow unsupervised-learning
Last synced: 01 Aug 2024
https://github.com/DeepVAC/deepvac
PyTorch Project Specification.
amp coreml ddp deepvac ncnn onnx python pytorch quantization tensorboard tensorrt torchscript
Last synced: 31 Jul 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: 31 Jul 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: 03 Aug 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: 31 Jul 2024
https://github.com/norse/norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
autograd deep-learning gpu machine-learning neural-network neuromorphic pytorch pytorch-lightning spiking-neural-networks tensor
Last synced: 24 Sep 2024
https://github.com/pdebench/pdebench
PDEBench: An Extensive Benchmark for Scientific Machine Learning
ai autoregressive-models benchmark deep-learning fluid-dynamics jax machine-learning navier-stokes-equations neural-networks neural-operators partial-differential-equations physics-informed-neural-networks pytorch scientific scientific-computing sciml simulation
Last synced: 01 Aug 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/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/msurtsukov/neural-ode
Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations
jupyter-notebook neural-ode pytorch vae
Last synced: 01 Aug 2024
https://github.com/pytorch/torcharrow
High performance model preprocessing library on PyTorch
Last synced: 24 Sep 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: 01 Aug 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: 01 Aug 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: 03 Aug 2024
https://github.com/BlackHC/tfpyth
Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).
machine-learning pytorch tensorflow
Last synced: 31 Jul 2024
https://github.com/iMoonLab/HGNN
Hypergraph Neural Networks (AAAI 2019)
complex-data deep-learning hypergraph hypergraph-neural-networks pytorch
Last synced: 31 Jul 2024
https://github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch
Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.
3d-human-pose cmu-panoptic computer-vision deep-learning human-pose-estimation keypoint-estimation lightweight multi-person-pose-estimation openvino pytorch real-time
Last synced: 02 Aug 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: 01 Aug 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: 27 Sep 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