Track objects in videos using SAM3

A Python library for amortized Bayesian workflows using generative neural networks.

Cambrian-S: Towards Spatial Supersensing in Video

Thermodynamic Hypergraphical Model Library in JAX

Official implementation of "Back to the Continuous Attractor" (NeurIPS 2024). This repository contains code for reproducing the experiments, pretrained models, and instructions for running the simu…

🤖 Just a command runner

Database with posteriors of interest for Bayesian inference

python package for approximately estimating mutual information in high dimensions

Probabilistic Programming and Nested sampling in JAX

Exact OU processes with JAX

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Access your entire server infrastructure from your local desktop

The official implementation of "MouseSIS: Space-Time Instance Segmentation of Mice" (ECCVW 2024)

Code for the paper "TreeLearn: A deep learning method for segmenting individual trees from ground-based LiDAR forest point clouds"

Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"

[NeurIPS 2021] Multiscale Benchmarks for Multimodal Representation Learning

Helpful tools and examples for working with flex-attention

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

NEural MOdelS, a statistical modeling framework for neuroscience.

A Python toolbox for analysing animal body movements across space and time

Schedule-Free Optimization in PyTorch

This repository contains code to download data for the preprint "MMEarth: Exploring Multi-Modal Pretext Tasks For Geospatial Representation Learning"

Uncertainty quantification with PyTorch

A generative AI extension for JupyterLab

TensorDict is a pytorch dedicated tensor container.

[pip install medmnist] 18x Standardized Datasets for 2D and 3D Biomedical Image Classification

State of the art inference for your bayesian models.