A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow

Official Jax Implementation of MD4 Masked Diffusion Models

Lifting ControlNet for Generalized Depth Conditioning

The official implementation of "[MASK] is All You Need"

Official repo for paper "Structured 3D Latents for Scalable and Versatile 3D Generation".

Mamba SSM architecture

ACL'2023: DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models

Tile primitives for speedy kernels

Fast and memory-efficient exact attention

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

Open roles at Kaedim and a summary of what it's like to work with us.

Official inference repo for FLUX.1 models

[CVPR 2025] 3DTopia-XL: High-Quality 3D PBR Asset Generation via Primitive Diffusion

📚 A collection of sketch based application papers.

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

[ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

Official codebase for "Schrodinger Bridges Beat Diffusion Models on Text-to-Speech Synthesis" (https://arxiv.org/abs/2312.03491).

Simplified Masked Diffusion Language Model

Supports SIGGRAPH Asia 2019 paper "GATA: Multi-Theme Generative Adversarial Terrain Amplification"

A unified framework for 3D content generation.

Common reusable tools

Source code repo for VectorFloorSeg: Two-Stream Graph Attention Network for Vectorized Roughcast Floorplan Segmentation

Code for paper "PuzzleFusion++: Auto-agglomerative 3D Fracture Assembly by Denoise and Verify" [ICLR2025]

An NVIDIA AI Workbench example project for an Agentic Retrieval Augmented Generation (RAG)

The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑‍🔬

Functionality-aware model evolution