AI Agent Skills

Siamese and triplet networks with online pair/triplet mining in PyTorch

Deep recommender models using PyTorch.

Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks…

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".

Convolutional Neural Network for 3D meshes in PyTorch

Ultralytics YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Instant neural graphics primitives: lightning fast NeRF and more

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

Lightning ⚡️ fast forecasting with statistical and econometric models.

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

Torchreid: Deep learning person re-identification in PyTorch.

Implementation of "BitNet: Scaling 1-bit Transformers for Large Language Models" in pytorch

Solve puzzles. Improve your pytorch.

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

Implementations of basic RL algorithms with minimal lines of codes! (pytorch based)