clumsy

NeMo

NeMo: a toolkit for conversational AI

aws-icons-for-plantuml

PlantUML sprites, macros, and other includes for Amazon Web Services services and resources

aws-neuron-sdk

Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

docker-py

A Python library for the Docker Engine API

dotfiles

My dotfiles

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

flet

Flet enables developers to easily build realtime web, mobile and desktop apps in Python. No frontend experience required.

furiosa

A Nord twist on the Furo theme

gloo

Collective communications library with various primitives for multi-machine training.

graphstorm

Enterprise graph machine learning framework for billion-scale graphs for ML scientists and data scientists.

helix

A post-modern modal text editor.

marimo

A reactive notebook for Python — run reproducible experiments, execute as a script, deploy as an app, and version with git.

Megatron-DeepSpeed

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

notebooks

plotlib

Data plotting library for Rust

pyarrow-stubs

Type annotations for pyarrow

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

rsrl

A fast, safe and easy to use reinforcement learning framework in Rust.

rusoto

AWS SDK for Rust

rustlings

:crab: Small exercises to get you used to reading and writing Rust code!

RustPython

A Python Interpreter written in Rust

sagemaker-python-sdk

A library for training and deploying machine learning models on Amazon SageMaker

sagemaker-training-toolkit

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

torchmetrics

Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.

torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

vision

Datasets, Transforms and Models specific to Computer Vision

xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)