Search.setIndex({"alltitles":{"A Simple Example":[[488,"a-simple-example"]],"Array":[[315,null]],"Attention layer":[[5,"attention-layer"]],"Automatic Differentiation":[[482,"automatic-differentiation"]],"Automatic Vectorization":[[482,"automatic-vectorization"]],"Basics":[[486,"basics"]],"Basics of Compile":[[480,"basics-of-compile"]],"Binary Size Minimization":[[8,"binary-size-minimization"]],"Binding to Python":[[2,"binding-to-python"]],"Build Options":[[8,"id3"]],"Build Requirements":[[8,"build-requirements"]],"Build and Install":[[8,null]],"Build from source":[[8,"build-from-source"]],"Building and Binding":[[2,"building-and-binding"]],"Building with CMake":[[2,"building-with-cmake"]],"Building with setuptools":[[2,"building-with-setuptools"]],"C++ API":[[8,"c-api"]],"C++ API Reference":[[7,null]],"Common Optimizers":[[474,null]],"Compilation":[[480,null]],"Compiling Training Graphs":[[480,"compiling-training-graphs"]],"Complex Example":[[1,"complex-example"]],"Conversion to NumPy and Other Frameworks":[[485,null]],"Converting the weights":[[5,"converting-the-weights"]],"Custom Extensions in MLX":[[2,null]],"Custom Metal Kernels":[[1,null]],"Data Types":[[316,null]],"Debugging":[[480,"debugging"]],"Devices and Streams":[[317,null]],"Differences from NumPy":[[483,"differences-from-numpy"]],"Distributed Communication":[[318,null],[481,null]],"Download the code":[[2,null],[5,null]],"Encoder layer":[[5,"encoder-layer"]],"Example Speedup":[[480,"example-speedup"]],"Examples":[[7,null]],"FFT":[[320,null]],"Fast":[[319,null]],"Full model":[[5,"full-model"]],"Function Transforms":[[482,null]],"Function and Graph Transformations":[[486,"function-and-graph-transformations"]],"Functions":[[449,null]],"Further Reading":[[7,null]],"Generation":[[5,"generation"]],"Getting Started":[[481,"getting-started"]],"Grid Sample VJP":[[1,"grid-sample-vjp"]],"Implementing the CPU Back-end":[[2,"implementing-the-cpu-back-end"]],"Implementing the GPU Back-end":[[2,"implementing-the-gpu-back-end"]],"Implementing the Primitive":[[2,"implementing-the-primitive"]],"Implementing the model":[[5,"implementing-the-model"]],"In Place Updates":[[483,"in-place-updates"]],"Indexing Arrays":[[483,null]],"Initializers":[[450,null]],"Inspecting Modules":[[323,"inspecting-modules"]],"Install":[[7,null]],"Installing MPI":[[481,"installing-mpi"]],"Introducing the Example":[[2,"introducing-the-example"]],"JAX":[[485,"jax"]],"LLM inference":[[5,null]],"Layers":[[451,null]],"Lazy Evaluation":[[484,null]],"Linear Algebra":[[321,null]],"Linear Regression":[[4,null]],"Loss Functions":[[452,null]],"MLX":[[7,null]],"Metal":[[322,null]],"Metal Debugger":[[3,null]],"Metal not found":[[8,"metal-not-found"]],"Module":[[453,null]],"Multi-Layer Perceptron":[[6,null]],"Neural Networks":[[323,null]],"Only Compute What You Use":[[484,"only-compute-what-you-use"]],"Operations":[[0,null],[2,"operations"],[454,null]],"Operations and Primitives":[[2,"operations-and-primitives"]],"Optimizer":[[475,null]],"Optimizers":[[455,null]],"Parameters":[[323,"parameters"]],"Primitive Transforms":[[2,"primitive-transforms"]],"Primitives":[[2,"primitives"]],"Pure Functions":[[480,"pure-functions"]],"Putting it all together":[[5,"putting-it-all-together"]],"PyTorch":[[485,"pytorch"]],"Python API":[[8,"python-api"]],"Python API Reference":[[7,null]],"Python Installation":[[8,"python-installation"]],"Quick Start Guide":[[486,null]],"Quick Start with Neural Networks":[[323,"quick-start-with-neural-networks"]],"Random":[[477,null]],"Results":[[2,"results"]],"Saving and Loading":[[455,"saving-and-loading"]],"Saving and Loading Arrays":[[487,null]],"Schedulers":[[476,null]],"Scripts":[[2,"scripts"],[5,"scripts"]],"Serialization Formats":[[487,"id1"]],"Setting up Remote Hosts":[[481,"setting-up-remote-hosts"]],"Si
