A library for efficient similarity search and clustering of dense vectors.

[NeurIPS 2024] Official PyTorch Implementation of PartCLIPSeg

(NeurIPS 2024 Oral 🔥) Improved Distribution Matching Distillation for Fast Image Synthesis

A 15TB Collection of Physics Simulation Datasets

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡

Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"

Vector (and Scalar) Quantization, in Pytorch

Cellular Automata Accelerated in JAX

Entropy Based Sampling and Parallel CoT Decoding

Any model. Any hardware. Zero compromise. Built with @ziglang / @openxla / MLIR / @bazelbuild

Perceptions of Probability and Numbers

Developer API service description and example client code

reproduction of semantic segmentation using masked autoencoder (mae)

Large-Scale Multimodal Dataset of Astronomical Data

3D Gaussian Splat Editor

This is the official code release for our work, Denoising Vision Transformers.

Cohere Toolkit is a collection of prebuilt components enabling users to quickly build and deploy RAG applications.

Official code repository for NeurIPS 2022 paper "SatMAE: Pretraining Transformers for Temporal and Multi-Spectral Satellite Imagery"

An instructive notebook implementation of anomaly detection algorithms, such as CATHODE, hiding the technical implementation details behind a scikit-learn-like API.

Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"

Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation

Fine-tune a 7B LLM on cosmology datasets

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)

T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!

A curated list of foundation models for vision and language tasks in medical imaging

SwinIR: Image Restoration Using Swin Transformer (official repository)

S2ORC: The Semantic Scholar Open Research Corpus: https://www.aclweb.org/anthology/2020.acl-main.447/

Sonification Tools and Resources for Analysis Using Sound Synthesis