Research Tools: code, datasets, & models

TorchGeo is a PyTorch (opens in new tab) domain library, similar to torchvision (opens in new tab), that provides datasets, transforms, samplers, and pre-trained models specific to geospatial data. The goal of this library is to make it…

Public release of sample videos to show the performance of a localization model (DIABLo). The model itself (or any related training or test data) will NOT be part of this release. We will only release…

Similar to the ID 3397 and ID 3860, the broad goal for this project is to research and develop Neuro-Symbolic systems for Natural Language Inferencing (NLI) to leverage the “correctness” guarantees and interpretability of symbolic…

HittER generates embeddings for large-scale knowledge graphs and performs link prediction using a hierarchical Transformer model. It appeared in EMNLP 2021.

Codebase for generative modeling of protein sequence and structure, including code for CNNs and GNNs and custom data handling code.

This repo contains the data and source code for baseline models in the NeurIPS 2021 benchmark paper for Constrained Language Understanding Evaluation Standard (CLUES) under MIT License.

AIDE: Annotation Interface for Data-driven Ecology – Tools for detecting wildlife in aerial images using active learning

Tutel MoE: An Optimized Mixture-of-Experts Implementation. Supported Framework: Pytorch Supported GPUs: CUDA(fp32 + fp16), ROCm(fp32 + fp16)

FOST(Forecasting open source tool) aims to provide an easy-use tool for spatial-temporal forecasting. The users only need to organize their data into a certain format and then get the prediction results with one command. FOST…

DiCE is a Python library that can generate counterfactual explanations for any machine learning classifier. Counterfactual explanations present “what-if” perturbations of the input such that an ML classifier outputs a different class for those perturbations…