Zum Hauptinhalt springen

Introduction

Welcome to Nix AI

Nix AI lets you train machine learning models by defining a single flake.nix configuration file that describes:

  • The Python dependencies required for your model
  • The datasets to be used in training
  • GPU configuration and training parameters
  • Output artifacts to preserve after training.

Nix and Nix AI​

Nix AI uses Nix to define the training environment configuration. Specifically, Nix AI uses the Nix programming language to describe training environments. Nix is a functional programming language similar to Haskell. The Nix package manager manages Python packages and system dependencies. Because Nix environments are reproducible and declarative, you can share your configuration file as part of your Git repository to ensure everyone who works on your project has identical training environments. You can learn about Nix in the Nix language tutorial.