LargeST: A Benchmark Dataset for Large-Scale Traffic Forecasting (NeurIPS 2023 DB Track)

Context geo-tiles in Python

Python tools for geographic data

A hyperparameter optimization framework

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

Time series forecasting with PyTorch

Code for "LoFTR: Detector-Free Local Feature Matching with Transformers", CVPR 2021, T-PAMI 2022

[CVPR 2024 Highlight] FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

PyTorch code and models for the DINOv2 self-supervised learning method.

Welcome to the project repository for POPE (Promptable Pose Estimation), a state-of-the-art technique for 6-DoF pose estimation of any object in any scene using a single reference.

An Open Platform for Population Synthesis

Synthetic populations from census data

Pytorch Implementation for Stepwise Goal-Driven Networks for Trajectory Prediction (RA-L/ICRA2022)

The code for Bi-directional Trajectory Prediction (BiTraP).

Annotation data for JAAD (Joint Attention in Autonomous Driving) Dataset

Annotations for Pedestrian Intention Estimation (PIE) dataset