GitHub repository: neverix/fae
Primary language: Python
Description: (jax, tpu) nf4 matmuls for flux + t5 + onnx vae. vision SAE training and maxacts