Example of a Convolutional Variational Autoencoder (CVAE) on MNIST (#264)

* initial commit

* style fixes

* update of ACKNOWLEDGMENTS

* fixed comment

* minor refactoring; removed unused imports

* added cifar and cvae to top-level README.md

* removed mention of cuda/mps in argparse

* fixed training status output

* load_weights() with strict=True

* pretrained model update

* fixed imports and style

* requires mlx>=0.0.9

* updated with results using mlx 0.0.9

* removed mention of private repo

* simplify and combine to one file, more consistency with other exmaples

* few more nits

* nits

* spell

* format

---------

Co-authored-by: Awni Hannun <awni@apple.com>

llms/mlx_lm/models/olmo.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from sys import exit
 from typing import Dict, Optional, Tuple, Union
 
 import mlx.core as mx
@@ -6,8 +7,6 @@ import mlx.nn as nn
 
 from .base import BaseModelArgs
 
-from sys import exit
-
 try:
     import hf_olmo
 except ImportError: