---
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
pipeline_tag: image-to-image
datasets:
- OpenTO/OpenTO
---

# Optimize Any Topology (OAT)

This repository contains the official implementation of **Optimize Any Topology (OAT)**, a foundation-model framework for shape- and resolution-free structural topology optimization. OAT directly predicts minimum-compliance layouts for arbitrary aspect ratios, resolutions, volume fractions, loads, and fixtures. It combines a resolution- and shape-agnostic autoencoder with an implicit neural-field decoder and a conditional latent-diffusion model.

![image](https://github.com/user-attachments/assets/6200fa2c-0cd5-49af-897c-67688f28c446)

**Paper**: [Optimize Any Topology: A Foundation Model for Shape- and Resolution-Free Structural Topology Optimization](https://huggingface.co/papers/2510.23667)
**Code**: https://github.com/ahnobari/OptimizeAnyTopology

## Abstract

Structural topology optimization (TO) is central to engineering design but remains computationally intensive due to complex physics and hard constraints. Existing deep-learning methods are limited to fixed square grids, a few hand-coded boundary conditions, and post-hoc optimization, preventing general deployment. We introduce Optimize Any Topology (OAT), a foundation-model framework that directly predicts minimum-compliance layouts for arbitrary aspect ratios, resolutions, volume fractions, loads, and fixtures. OAT combines a resolution- and shape-agnostic autoencoder with an implicit neural-field decoder and a conditional latent-diffusion model trained on OpenTO, a new corpus of 2.2 million optimized structures covering 2 million unique boundary-condition configurations. On four public benchmarks and two challenging unseen tests, OAT lowers mean compliance up to 90% relative to the best prior models and delivers sub-1 second inference on a single GPU across resolutions from 64 x 64 to 256 x 256 and aspect ratios as high as 10:1. These results establish OAT as a general, fast, and resolution-free framework for physics-aware topology optimization and provide a large-scale dataset to spur further research in generative modeling for inverse design. Code & data can be found at this https URL.

## The Open-TO Dataset

The OpenTO dataset, which was used for training OAT, is publicly available on the Hugging Face Hub at [OpenTO/OpenTO](https://huggingface.co/datasets/OpenTO/OpenTO).

## Pretrained Checkpoints

Pretrained checkpoints for both the Neural Field Auto-Encoder (NFAE) and Latent Diffusion Model (LDM) components of OAT are available on the Hugging Face Hub:

-   Auto Encoder: [OpenTO/NFAE](https://huggingface.co/OpenTO/NFAE)
-   Latent Diffusion: [OpenTO/LDM](https://huggingface.co/OpenTO/LDM)
-   Auto Encoder Large Latent: [OpenTO/NFAE_L](https://huggingface.co/OpenTO/NFAE_L)
-   Latent Diffusion Large Latent: [OpenTO/LDM_L](https://huggingface.co/OpenTO/LDM_L)