User Guide
API Documentation
Examples & Tutorials
Development
NewGELU
NewGELU.forward()
CausalSelfAttention
CausalSelfAttention.__init__()
CausalSelfAttention.forward()
Block
Block.__init__()
Block.forward()
VectorQuantizer
VectorQuantizer.__init__()
VectorQuantizer.forward()
VectorQuantizerEMA
VectorQuantizerEMA.__init__()
VectorQuantizerEMA.forward()
Residual
Residual.__init__()
Residual.forward()
ResidualStack
ResidualStack.__init__()
ResidualStack.forward()
Encoder
Encoder.__init__()
Encoder.forward()
Jitter
Jitter.__init__()
Jitter.forward()
Decoder
Decoder.__init__()
Decoder.forward()
VQVAE
VQVAE.__init__()
VQVAE.forward()