ahmedgh970
Follow
Stars
Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"
Transformer: PyTorch Implementation of "Attention Is All You Need"
Official PyTorch Implementation of MambaVision: A Hybrid Mamba-Transformer Vision Backbone
ScenarioNet: Scalable Traffic Scenario Management System for Autonomous Driving
A unified interface to many trajectory forecasting datasets.
[CVPR 2024] Test-Time Training of Trajectory Prediction via Masked Autoencoder and Actor-specific Token Memory
A Unified Framework for scalable Vehicle Trajectory Prediction
[CVPR 2024 Award Candidate] Producing and Leveraging Online Map Uncertainty in Trajectory Prediction
Count the MACs / FLOPs of your PyTorch model.
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
MambaOut: Do We Really Need Mamba for Vision?
A tool to download waymo open dataset (https://github.com/waymo-research/waymo-open-dataset) and convert it from TFrecord to general format
High-Resolution Image Synthesis with Latent Diffusion Models
ILVR: Conditioning Method for Denoising Diffusion Probabilistic Models (ICCV 2021 Oral)
PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
Video generation tool for Stable Diffusion.
📚 A collection of edge/contour/boundary detection papers and toolbox.
Zero-Shot Video Editing Using Off-The-Shelf Image Diffusion Models
Computation-Efficient Era: A Comprehensive Survey of State Space Models in Medical Image Analysis
Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)
Language Modeling with the H3 State Space Model