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
434 forks
Language: C++
last commit: 6 days ago

Related projects:

Repository Description Stars
mlflow/mlflow A platform to manage the entire machine learning lifecycle, from experiment tracking to model deployment. 18,781
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,035
aksnzhy/xlearn A high-performance machine learning package with linear models and factorization machines. 3,087
mlc-ai/mlc-llm Enables the development, optimization, and deployment of large language models on various platforms using a unified high-performance inference engine. 19,197
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,296
google/cfu-playground A framework for designing and optimizing machine learning accelerators on FPGAs. 472
dotnet/machinelearning-samples A collection of samples and examples demonstrating the usage of ML.NET for machine learning tasks in .NET applications. 4,490
pytorch/glow A compiler and execution engine for neural networks that generates optimized code for hardware accelerators 3,235
openmathlib/openblas An optimized BLAS library based on GotoBLAS2 6,392
lightning-ai/lit-llama An implementation of a large language model using the nanoGPT architecture 5,993
microsoft/synapseml A library for building scalable machine learning pipelines on distributed computing frameworks like Apache Spark 5,068
fedml-ai/fedml A unified and scalable machine learning library for large-scale distributed training, model serving, and federated learning. 4,187
ludwig-ai/ludwig A low-code framework for building custom deep learning models and neural networks 11,189
mlpack/mlpack A C++ machine learning library with bindings to other languages and bindings for multiple programming languages. 5,113
tensorflow/tfjs An open-source JavaScript library for training and deploying machine learning models using WebGL acceleration. 18,495