Streamlines and simplifies prompt design for both developers and non-technical users with a low code approach.

A native PyTorch Library for large model training

Ansible Role to install RKE2 Kubernetes.

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.

Monte Carlo tree search in JAX

A tiny but valid `init` for containers

NGINX S3 Caching Gateway

ChatGPT for Mac, living in your menubar.

Open Source Feature Flagging and A/B Testing Platform

🦜🔗 Build context-aware reasoning applications

Amplify your team's potential with customizable and secure AI assistants.

SkyPilot: Run AI and batch jobs on any infra (Kubernetes or 12+ clouds). Get unified execution, cost savings, and high GPU availability via a simple interface.

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

Karras et al. (2022) diffusion models for PyTorch

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure as a dataclass and let pyrallis do the rest!

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

Transformer related optimization, including BERT, GPT

Repo for external large-scale work

Multi-GPU CUDA stress test

Timeout decorator for Python

A subprocess.run drop-in replacement that supports a tee mode, being able to display output in real time while still capturing it. No dependencies needed

functorch is JAX-like composable function transforms for PyTorch.

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

A specification that python filesystems should adhere to.

Ongoing research training transformer language models at scale, including: BERT & GPT-2

Ongoing research training transformer language models at scale, including: BERT & GPT-2

a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.