Add llms subdir + update README (#145)

* add llms subdir + update README

* nits

* use same pre-commit as mlx

* update readmes a bit

* format

mnist/jax_main.py

@@ -1,9 +1,10 @@
 # Copyright © 2023 Apple Inc.
 
+import functools
+import time
+
 import jax
 import jax.numpy as jnp
-import functools
-import time
 
 import mnist
 

mnist/main.py

@@ -3,11 +3,10 @@
 import argparse
 import time
 
-import numpy as np
-
 import mlx.core as mx
 import mlx.nn as nn
 import mlx.optimizers as optim
+import numpy as np
 
 import mnist
 

mnist/mnist.py

@@ -1,11 +1,12 @@
 # Copyright © 2023 Apple Inc.
 
 import gzip
-import numpy as np
 import os
 import pickle
 from urllib import request
 
+import numpy as np
+
 
 def mnist(save_dir="/tmp"):
     """

mnist/torch_main.py

@@ -1,9 +1,10 @@
 # Copyright © 2023 Apple Inc.
 
 import argparse
-import torch
 import time
 
+import torch
+
 import mnist