optimizer based DeepSpeed implementation for diffusion policy.

Input

4232-dim embedding

Encoder

110 x Transformer with 18 heads

Output

perplexity projection

spice based TensorFlow implementation for peft layer.

Input

1593-dim embedding

Encoder

117 x MLP with 46 heads

Output

precision projection