An open API service indexing awesome lists of open source software.

https://github.com/jxtngx/tunelab

A web app for post-training
https://github.com/jxtngx/tunelab

artificial-intelligence fastapi generative-ai nextjs python pytorch shadcn-ui typescript

Last synced: 3 months ago
JSON representation

A web app for post-training

Awesome Lists containing this project

README

          

# Tune Lab

Tune Lab is a [FastAPI](https://fastapi.tiangolo.com) + [NextJS](https://nextjs.org) application that is purpose built for post-training with [NVIDIA NeMo](https://github.com/NVIDIA/NeMo) and [HuggingFace TRL](https://github.com/huggingface/trl) on MacOS + MPS, NVIDIA DGX devices + CUDA, and CUDA compatible AWS EC2 DLAMI instances.

> [!IMPORTANT]
> Tune Lab is in active development and is not yet ready for use

> [!IMPORTANT]
> Tune Lab is built alongside [Cursor](https://cursor.com/) and [Anthropic's](https://www.anthropic.com/) Claude and is closed for contributions

## Intended Use

Tune Lab is intended to be cloned and used locally on MacOS or NVIDIA DGX devices, or hosted on AWS.

> [!IMPORTANT]
> Cloning the repository does not entitle a user to support from the maintainer

## High-Level Features

- a RESTful API for managing and tuning models with NeMo and TRL
- a NextJS web interface for managing and tuning models
- a Cursor directory for advanced workflows across finetuning, backend infra, and frontend development

## Downstream Usage

Models tuned with Tune Lab are intended to be used with LangChain or AWS Strands or NVIDIA NeMo Agent Toolkit.