Stars
A minimal Python logger that tracks everything you try when building AI - metrics, prompts, models, etc, so you can see what changed and why.
A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.
PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
Speed up model training by fixing data loading.
Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.
Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.
A gtk-rs app for interacting with Nvidia GPU management interfaces like nvidia-settings and nvidia-smi.
Demos from my Machine Learning talk at OpenFest 2022
Learn to serve Stable Diffusion models on cloud infrastructure at scale. This Lightning App shows load-balancing, orchestrating, pre-provisioning, dynamic batching, GPU-inference, micro-services wo…
CAIRI Supervised, Semi- and Self-Supervised Visual Representation Learning Toolbox and Benchmark
A curated list of examples that use Pachyderm to accomplish various tasks.
Common Python utilities and GitHub Actions in Lightning Ecosystem
Connects Nuclio / CVAT to Lighting flash ML models
A sample app which uses Monaco Editor integration, and runs the script on frame from webcam
Perform Image/Text Classification with ⚡ Flash built using ⚡ AI
Simple tooling for instance segmentation aiming at cell biology...
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
A collection of libraries to optimise AI model performances
GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compression of numerical and other data types in HPC/ML applications.
Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research techniques without significant engineering overhead.Specifi…
Domain adaptation made easy. Fully featured, modular, and customizable.
The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
Accelerated pose estimation and tracking using semi-supervised learning
Automate issue discovery for your projects against Lightning nightly and releases.
A unified framework for Deep Learning Models on tabular data
Time series forecasting with PyTorch
Practical Deep Learning at Scale with MLFlow, published by Packt