All Examples¤

Quick Start¤

This catalog includes four published example buckets: runtime-backed Artifex tutorials, standalone concept walkthroughs, exploratory workflows, and validation utilities. Runtime-backed tutorials instantiate shipped Artifex owners end to end. Standalone concept walkthroughs are clearly labeled raw JAX/Flax NNX pedagogy that do not instantiate shipped Artifex runtime owners. Exploratory workflows keep useful lower-level or partially retained material visible without presenting it as canonical. Validation utilities are quick environment or technology-stack checks rather than model tutorials. Still-relevant topics without runnable pairs live in Planned Example Topics.

VAE on MNIST

Beginner

Train a Variational Autoencoder on MNIST digits

VAE Image MNIST

⏱ 10 min

GAN on MNIST

Beginner

Train a Generative Adversarial Network on MNIST

GAN DCGAN MNIST

⏱ 15 min

Diffusion on MNIST

Beginner

Train a DDPM diffusion model on MNIST digits

Diffusion DDPM MNIST

⏱ 15 min

Diffusion MNIST Training

Beginner

Complete training workflow for diffusion models on MNIST

Diffusion Training MNIST

⏱ 20 min

Flow on MNIST

Beginner

Train a normalizing flow model on MNIST

Flow RealNVP MNIST

⏱ 10 min

Diffusion Models¤

DiT Demo

Advanced

Scalable diffusion with Diffusion Transformers

DiT Transformer Scalable

⏱ 30 min

Advanced Tutorials¤

Advanced Training Pipeline

Intermediate

Production-ready training with optimizers, schedulers, and checkpointing

Training Optimizer Scheduler Checkpointing

⏱ 2 min

Advanced VAE

Advanced

β-VAE, VQ-VAE, and disentanglement techniques

β-VAE VQ-VAE Disentanglement

⏱ 45 min

Specialized Models¤

Energy-Based Model

Advanced

EBM training with MCMC and Langevin sampling

EBM MCMC Langevin

⏱ 25 min

Geometric Benchmark

Advanced

Geometric generative model benchmarking

3D Point Cloud Mesh

⏱ 45 min

Geometric Models¤

Geometric Models Demo

Beginner

Quick reference for point clouds, meshes, and voxels

Point Cloud Mesh Voxel

⏱ 5 min

Geometric Losses Demo

Intermediate

Loss functions for point clouds, meshes, and voxels

Loss Functions Point Cloud Mesh Voxel

⏱ 15 min

Simple Point Cloud

Beginner

Generate and visualize 3D point clouds with transformers

Point Cloud Transformer 3D

⏱ 10 min

Geometric Benchmark

Advanced

Complete evaluation on geometric tasks

Benchmark ShapeNet Metrics

⏱ 45 min

Protein Modeling¤

Protein Extensions

Intermediate

Add domain-specific constraints with protein extensions

Extensions Constraints Bonds

⏱ 10 min

Protein Model Extension

Intermediate

Combine point cloud models with protein extensions

Protein Extensions Point Cloud

⏱ 10 min

Protein Model with Modality

Intermediate

Using the modality architecture for protein models

Protein Modality Factory

⏱ 10 min

Protein Point Cloud

Intermediate

Point cloud modeling with geometric constraints

Protein Point Cloud Constraints

⏱ 15 min

Protein Extensions

Intermediate

Using protein extensions with configuration system

Extensions Config Protein

⏱ 10 min

Protein-Ligand Benchmark

Advanced

Protein-ligand binding site generation

Protein SE(3) Equivariant

⏱ 60 min

Framework & Techniques¤

BlackJAX Integration

Intermediate

MCMC sampling with BlackJAX: HMC, NUTS, and MALA algorithms

MCMC HMC NUTS BlackJAX

⏱ 15 min

BlackJAX Sampling Examples

Advanced

Compare HMC, MALA, NUTS samplers and direct BlackJAX API usage

MCMC Comparison Performance

⏱ 20 min

BlackJAX Integration Examples

Advanced

Direct API vs functional API: progress bars, JIT compilation, and performance

Integration Performance JIT

⏱ 25 min

Loss Functions

Intermediate

Complete guide to Artifex loss functions

Losses KL Adversarial

⏱ 20 min

Framework Features

Intermediate

Explore Artifex's architectural patterns

Architecture Patterns Framework

⏱ 30 min

β-VAE Benchmark

Advanced

Compare β-VAE configurations and disentanglement

β-VAE Benchmark Tuning

⏱ 40 min

Exploratory Workflows¤

Advanced GAN

Advanced

Exploratory comparison of lower-level GAN building blocks and custom training loops

Exploratory GAN Lower-Level

⏱ 45 min

Protein Diffusion

Advanced

Exploratory direct-owner workflow for protein point-cloud and graph model surfaces

Exploratory Protein Direct Owners

⏱ 15 min

Validation Utilities¤

Protein Tech Validation

Beginner

Validation utility for JAX/Flax NNX environment checks around protein experimentation

Validation Protein Setup

⏱ 5 min

Standalone Concept Walkthroughs¤

Simple Diffusion

Intermediate

Standalone JAX/Flax NNX walkthrough of diffusion denoising concepts

Standalone Diffusion Image

⏱ 20 min

Audio Generation

Intermediate

Standalone waveform-generation walkthrough with raw NNX layers

Standalone Audio Waveform

⏱ 30 min

Simple Text Generation

Beginner

Standalone character-level text generation walkthrough

Standalone Text Character-Level

⏱ 5 min

Simple Image-Text Multimodal

Intermediate

Standalone image-text retrieval walkthrough with local encoders

Standalone Multimodal Retrieval

⏱ 10 min

Reference Tables¤

📊 Browse by Category

By Model Type¤

Model	Examples	Level
VAE	Basic, Advanced, β-VAE Benchmark	⭐ - ⭐⭐⭐
GAN	Basic, Exploratory Advanced Workflow	⭐ - ⭐⭐⭐
Diffusion	Basic, Training, Standalone Simple, DiT	⭐ - ⭐⭐⭐
Flow	Basic	⭐ - ⭐⭐
EBM	Simple EBM	⭐⭐⭐
Text	Standalone Simple Text Generation	⭐
Multimodal	Standalone Image-Text	⭐⭐
Protein	Modality, Point Cloud, Extensions, Ligand, Exploratory Protein Diffusion, Validation Tech Check	⭐⭐ - ⭐⭐⭐
Geometric	Benchmark	⭐⭐⭐

By Dataset¤

Dataset	Examples
MNIST	VAE · GAN · Diffusion · Diffusion Training · Flow
Text	Standalone Simple Text Generation
Multimodal	Standalone Image-Text
Audio	Standalone Audio Generation
Protein/3D	Modality · Point Cloud · Extensions · Ligand · Exploratory Protein Diffusion · Validation Tech Check · Geometric