amazon-science

Chronos: Pretrained Models for Time Series Forecasting

Official implementation for "Multimodal Chain-of-Thought Reasoning in Language Models" (stay tuned and more will be updated)

Official implementation for "Automatic Chain of Thought Prompting in Large Language Models" (stay tuned & more will be updated)

RAGChecker: A Fine-grained Framework For Diagnosing RAG

SiamMOT: Siamese Multi-Object Tracking

Official implementation of Earthformer

RefChecker provides automatic checking pipeline and benchmark dataset for detecting fine-grained hallucinations generated by Large Language Models.

Shopping Queries Dataset: A Large-Scale ESCI Benchmark for Improving Product Search

BigDetection: A Large-scale Benchmark for Improved Object Detector Pre-training

Pytorch implementation of Supporting Clustering with Contrastive Learning, NAACL 2021

Repository for Fraud Dataset Benchmark

Project for <SPot-the-Difference Self-Supervised Pre-training for Anomaly Detection and Segmentation> (ECCV 2022)

Official PyTorch implementation of TSDiff models presented in the NeurIPS 2023 paper "Predict, Refine, Synthesize: Self-Guiding Diffusion Models for Probabilistic Time Series Forecasting"

Official implementation for the paper "Prompt Pre-Training with Over Twenty-Thousand Classes for Open-Vocabulary Visual Recognition"

ReFinED is an efficient and accurate entity linking (EL) system.

Official Implementations of "Mixed-Type Tabular Data Synthesis with Score-based Diffusion in Latent Space""

AuctionGym is a simulation environment that enables reproducible evaluation of bandit and reinforcement learning methods for online advertising auctions.

CrossCodeEval: A Diverse and Multilingual Benchmark for Cross-File Code Completion (NeurIPS 2023)

JuLS is a Julia Local Search solver that combines Constraint Based Local Search (CBLS) and Constraint Programming (CP)

Video Contrastive Learning with Global Context, ICCVW 2021

[NeurIPS 2021 Spotlight] Official implementation of Long Short-Term Transformer for Online Action Detection

This package provides a pythorch implementation of "GAN-Control: Explicitly Controllable GANs", ICCV 2021.

Structured Prediction as Translation between Augmented Natural Languages

Official implementation for Training LLMs with MXFP4

CrossNorm and SelfNorm for Generalization under Distribution Shifts, ICCV 2021

MixGen: A New Multi-Modal Data Augmentation

CodeSage: Code Representation Learning At Scale (ICLR 2024)

Dataset from the paper "Mintaka: A Complex, Natural, and Multilingual Dataset for End-to-End Question Answering" (COLING 2022)

This repo provides code and data used in our TANDA paper.

Code for the paper "Meta-Q-Learning"( ICLR 2020)

PyTorch implementation of EMAN for self-supervised and semi-supervised learning: https://arxiv.org/abs/2101.08482

Official implementation for "GLASS: Global to Local Attention for Scene-Text Spotting" (ECCV'22)

Cyber-Zero: Training Cybersecurity Agents Without Runtime

This is an official implementation of TubeR: Tubelet Transformer for Video Action Detection

Code related to "Have Your Text and Use It Too! End-to-End Neural Data-to-Text Generation with Semantic Fidelity" paper

Code repo for the ICML 2024 paper "Automated Evaluation of Retrieval-Augmented Language Models with Task-Specific Exam Generation"

Dataset associated with "BOLD: Dataset and Metrics for Measuring Biases in Open-Ended Language Generation" paper

SWE-PolyBench: A multi-language benchmark for repository level evaluation of coding agents

Multimodal Semi-Supervised Learning for Text Recognition (SemiMTR)

Official implementation of the CVPR'24 paper [Adaptive Slot Attention: Object Discovery with Dynamic Slot Number]

Redset is a dataset containing three months worth of user query metadata that ran on a selected sample of instances in the Amazon Redshift fleet. We provide query metadata for 200 provisioned and...

Training Language Model Agents to Find Vulnerabilities with CTF-Dojo

A dataset to facilitate the research of Structure-from-Motion (SfM) for movie and TV shows.

PyTorch implementation of Omni-DETR for omni-supervised object detection: https://arxiv.org/abs/2203.16089

Repoformer: Selective Retrieval for Repository-Level Code Completion (ICML 2024)

Progressive Coordinate Transforms for Monocular 3D Object Detection, NeurIPS 2021

PyTorch implementation of Semi-supervised Vision Transformers

AgentOccam: A Simple Yet Strong Baseline for LLM-Based Web Agents

CrossCLR: Cross-modal Contrastive Learning For Multi-modal Video Representations, ICCV 2021

Code for EmBERT, a transformer model for embodied, language-guided visual task completion.

CaML: Carbon Footprinting of Household Products with Zero-Shot Semantic Text Similarity

Official implementation for the paper "A Cheaper and Better Diffusion Language Model with Soft-Masked Noise"

Releasing code for "ReCode: Robustness Evaluation of Code Generation Models"

Official Implementation for ICCV'23 paper Coarse-to-Fine Amodal Segmentation with Shape Prior (C2F-Seg).

Research code and optimized inference for ALoDLM: Adaptively Looped Diffusion Language Models.

Datasets and code for results presented in the ProbConserv paper

Code from the paper "What do Models Learn from Question Answering Datasets?" (EMNLP 2020)

Implementation of the paper "FactGraph: Evaluating Factuality in Summarization with Semantic Graph Representations (NAACL 2022)"

Code associated with the "Data Augmentation using Pre-trained Transformer Models" paper