Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python

Official code for "FeatUp: A Model-Agnostic Frameworkfor Features at Any Resolution" ICLR 2024

Testing adaptation of the DINOv2 encoder for vision tasks with Low-Rank Adaptation (LoRA)

PyTorch code and models for the DINOv2 self-supervised learning method.

Mamba SSM architecture

Multi functional app to find duplicates, empty folders, similar images etc.

Perform data science on data that remains in someone else's server

Efficient Distributed GPU Programming for Exascale, an SC/ISC Tutorial

Tutorial on Accelerating Generative AI with PyTorch at ISC 2024

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

Coz: Causal Profiling

An official implementation of Pangu-Weather

An Open-source Framework for Scalable Neural Networks on Heterogeneous Systems

Structured Matrix Package (LBNL)

ænet-PyTorch: a GPU-supported implementation for machine learning atomic potentials training

COYO-700M: Large-scale Image-Text Pair Dataset

An Aspiring Drop-In Replacement for NumPy at Scale

Test suite for probing the numerical behavior of NVIDIA tensor cores

Livermore Big Artificial Neural Network Toolkit

Contains the material for the Machine Learning on HPC systems course on 16-05-2023

Now hosted on GitLab.

Interact with your documents using the power of GPT, 100% privately, no data leaks

Machine Learning for HPC Workflows

Generation of diagrams like flowcharts or sequence diagrams from text in a similar manner as markdown

🦜🔗 Build context-aware reasoning applications

A python package that brings support for reading, writing and converting GPX files.

🐙 Guides, papers, lecture, notebooks and resources for prompt engineering