xla

ML compiler

Compiles machine learning models for high-performance execution on various hardware platforms

A machine learning compiler for GPUs, CPUs, and ML accelerators

GitHub

3k stars
41 watching
451 forks
Language: C++
last commit: about 2 months ago

Related projects:

Repository Description Stars
mlflow/mlflow A platform for managing machine learning projects from inception to deployment 19,021
dotnet/machinelearning A cross-platform machine learning framework for .NET that enables developers to build, train, and deploy models without prior expertise in ML. 9,071
aksnzhy/xlearn A high-performance machine learning package with linear models and factorization machines. 3,087
mlc-ai/mlc-llm A machine learning compiler and deployment engine for large language models 19,396
ml-explore/mlx A high-performance machine learning framework designed by researchers for machine learning research, providing an efficient and user-friendly platform for training and deploying models. 17,878
google/cfu-playground A framework for designing and evaluating custom processor instructions to accelerate machine learning tasks on FPGAs. 476
dotnet/machinelearning-samples A collection of samples and examples demonstrating the usage of ML.NET for machine learning tasks in .NET applications. 4,508
pytorch/glow A compiler and execution engine for neural networks that generates optimized code for hardware accelerators 3,247
openmathlib/openblas An optimized BLAS library based on GotoBLAS2 for high-performance linear algebra operations. 6,453
lightning-ai/lit-llama An implementation of a large language model using the nanoGPT architecture 6,013
microsoft/synapseml A library for building scalable machine learning pipelines on distributed computing frameworks like Apache Spark 5,083
fedml-ai/fedml A unified and scalable machine learning library for large-scale distributed training, model serving, and federated learning. 4,205
ludwig-ai/ludwig A low-code framework for building custom deep learning models and neural networks 11,236
mlpack/mlpack A C++ machine learning library with bindings to other languages and bindings for multiple programming languages. 5,151
tensorflow/tfjs An open-source JavaScript library for training and deploying machine learning models using WebGL acceleration. 18,533