Search.setIndex({"alltitles":{"A Simple Example":[[522,"a-simple-example"]],"Array":[[343,null]],"Attention layer":[[6,"attention-layer"]],"Automatic Differentiation":[[515,"automatic-differentiation"]],"Automatic Vectorization":[[515,"automatic-vectorization"]],"Basics":[[520,"basics"]],"Basics of Compile":[[512,"basics-of-compile"]],"Basics of Exporting":[[514,"basics-of-exporting"]],"Binary Size Minimization":[[9,"binary-size-minimization"]],"Binding to Python":[[2,"binding-to-python"]],"Build Options":[[9,"id5"]],"Build Requirements":[[9,"build-requirements"]],"Build and Install":[[9,null]],"Build from source":[[9,"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":[[9,"c-api"]],"C++ API Reference":[[8,null]],"CPU-only (Linux)":[[9,"cpu-only-linux"]],"CUDA":[[9,"cuda"],[9,"id3"]],"Common Optimizers":[[506,null]],"Compilation":[[512,null]],"Compiling Training Graphs":[[512,"compiling-training-graphs"]],"Complex Example":[[1,"complex-example"]],"Conversion to NumPy and Other Frameworks":[[519,null]],"Converting the weights":[[6,"converting-the-weights"]],"Custom Extensions in MLX":[[2,null]],"Custom Metal Kernels":[[1,null]],"Data Types":[[344,null]],"Debugging":[[512,"debugging"]],"Defining a Ring":[[513,"defining-a-ring"]],"Devices and Streams":[[345,null]],"Differences from NumPy":[[516,"differences-from-numpy"]],"Distributed Communication":[[346,null],[513,null]],"Download the code":[[2,null],[6,null]],"Encoder layer":[[6,"encoder-layer"]],"Example Speedup":[[512,"example-speedup"]],"Examples":[[8,null]],"Export Functions":[[347,null]],"Exporting Functions":[[514,null]],"Exporting Modules":[[514,"exporting-modules"]],"Exporting Multiple Traces":[[514,"exporting-multiple-traces"]],"FFT":[[349,null]],"Fast":[[348,null]],"Full model":[[6,"full-model"]],"Function Transforms":[[515,null]],"Function and Graph Transformations":[[520,"function-and-graph-transformations"]],"Functions":[[479,null]],"Further Reading":[[8,null]],"Generation":[[6,"generation"]],"Getting Started":[[513,"getting-started"]],"Getting Started with MPI":[[513,"getting-started-with-mpi"]],"Getting Started with Ring":[[513,"getting-started-with-ring"]],"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":[[6,"implementing-the-model"]],"Importing Functions in C++":[[514,"importing-functions-in-c"]],"In Place Updates":[[516,"in-place-updates"]],"Indexing Arrays":[[516,null]],"Initializers":[[480,null]],"Inspecting Modules":[[353,"inspecting-modules"]],"Install":[[8,null]],"Installing MPI":[[513,"installing-mpi"]],"Introducing the Example":[[2,"introducing-the-example"]],"JAX":[[519,"jax"]],"LLM inference":[[6,null]],"Launching Distributed Programs":[[517,null]],"Layers":[[481,null]],"Lazy Evaluation":[[518,null]],"Linear Algebra":[[350,null]],"Linear Regression":[[5,null]],"Linux":[[9,"linux"]],"Loss Functions":[[482,null]],"MLX":[[8,null]],"MPI Specifics":[[517,"mpi-specifics"]],"Memory Management":[[351,null]],"Metal":[[352,null]],"Metal Debugger":[[3,null]],"Metal not found":[[9,"metal-not-found"]],"Module":[[483,null]],"More Examples":[[514,"more-examples"]],"Multi-Layer Perceptron":[[7,null]],"Neural Networks":[[353,null]],"Only Compute What You Use":[[518,"only-compute-what-you-use"]],"Operations":[[0,null],[2,"operations"],[484,null]],"Operations and Primitives":[[2,"operations-and-primitives"]],"Optimizer":[[507,null]],"Optimizers":[[485,null]],"Package Variables":[[4,"id1"]],"Parameters":[[353,"parameters"]]
