Stars
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
A PyTorch native platform for training generative AI models
Efficient Triton Kernels for LLM Training
Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
DeepGNN is a framework for training machine learning models on large scale graph data.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Dynamic, resilient AI orchestration. Coordinate data, models, and compute as you build AI workflows. Flyte 2 now available locally: https://github.com/flyteorg/flyte-sdk
Model Registry provides a single pane of glass for ML model developers to index and manage models, versions, and ML artifacts metadata. It fills a gap between model experimentation and production a…
Apache Airflow - A platform to programmatically author, schedule, and monitor workflows
Caikit is an AI toolkit that enables users to manage models through a set of developer friendly APIs.
Interpretability and explainability of data and machine learning models
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Source code for the X Recommendation Algorithm
A conda-smithy repository for kfp-tekton.
Curated list of awesome tools for managing open source programs
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
A toolkit to run Ray applications on Kubernetes
Elyra extends JupyterLab with an AI centric approach.
A multi-cloud framework for big data analytics and embarrassingly parallel jobs, that provides an universal API for building parallel applications in the cloud ☁️🚀
Bootstrap Kubernetes the hard way. No scripts.
Machine Learning eXchange (MLX). Data and AI Assets Catalog and Execution Engine
Official electron build of draw.io
A kubernetes based framework for hassle free handling of datasets
Experimental project plugging Tekton yaml behind KFP API and UI engine