vturrisi
Follow
Lists (1)
Starred repositories
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
Collection of kernels written in Triton language
Getting crystal-like representations with harmonic loss
PyTorch native quantization and sparsity for training and inference
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)
Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
Vector (and Scalar) Quantization, in Pytorch
Mixbox is a library for natural color mixing based on real pigments.
Official implementation for paper "Knowledge Diffusion for Distillation", NeurIPS 2023
PyTorch code for hierarchical k-means -- a data curation method for self-supervised learning
⚡ TabPFN: Foundation Model for Tabular Data ⚡
Official inference repo for FLUX.1 models
[ECCV 2024] Official PyTorch implementation code for realizing the technical part of Mixture of All Intelligence (MoAI) to improve performance of numerous zero-shot vision language tasks.
Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.
Efficient Triton Kernels for LLM Training
High-resolution models for human tasks.
An extremely fast Python package and project manager, written in Rust.
A modular graph-based Retrieval-Augmented Generation (RAG) system
A PyTorch native library for large model training
This repository is a curated collection of the most exciting and influential CVPR 2024 papers. 🔥 [Paper + Code + Demo]
Typer, build great CLIs. Easy to code. Based on Python type hints.
Recipes for shrinking, optimizing, customizing cutting edge vision models. 💜
An efficient pure-PyTorch implementation of Kolmogorov-Arnold Network (KAN).
