machineko

SwiftCU

SwiftCU is a wrapper for CUDA runtime API's (exposed as cxxCU) with extra utilities for device management, memory ops and kernel execution, along with a robust suite of tests. Repo is tested on newest (v12.5) CUDA runtime API on both Linux and Windows.

SwiftCUBLAS

SwiftCUBLAS is a wrapper for cuBLAS APIs with extra utilities for ease of usage, along with a suite of tests. The repository is tested on the newest (v12.5) CUDA runtime API on both Linux and Windows.

SwiftCUBLASLt

ConvNeXt-V2

Code release for ConvNeXt V2 model

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

BrushNet

[ECCV 2024] The official implementation of paper "BrushNet: A Plug-and-Play Image Inpainting Model with Decomposed Dual-Branch Diffusion"

davis-interactive

Package to evaluate interactive segmentation with scribbles at DAVIS dataset.

DiffusionLight

[CVPR 2024] code release for "DiffusionLight: Light Probes for Free by Painting a Chrome Ball"

dinov2

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

EfficientFormer

flatbuffers

FlatBuffers: Memory Efficient Serialization Library

flaxmodels

Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

HashNeRF-pytorch

Pure PyTorch Implementation of NVIDIA paper on Instant Training of Neural Graphics primitives: https://nvlabs.github.io/instant-ngp/

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

kickstart.nvim

A launch point for your personal nvim configuration

kornia

Geometric Computer Vision Library for Spatial AI

MiVOS

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!

ml-stable-diffusion

Stable Diffusion with Core ML on Apple Silicon

mlx

MLX: An array framework for Apple silicon

nerf-pytorch

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

NeuS-FIX

Code release for NeuS

oidn

Intel® Open Image Denoise library

PixArt-alpha

PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis

PlanarReconstruction

[CVPR'19] Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding

PowerPaint

[ECCV 2024] PowerPaint, a versatile image inpainting model that supports text-guided object inpainting, object removal, image outpainting and shape-guided object inpainting with only a single model. 一个高质量多功能的图像修补模型，可以同时支持插入物体、移除物体、图像扩展、形状可控的物体生成，只需要一个模型

PythonKit-GIL

Swift framework to interact with Python.

StrDiffusion

[CVPR 2024] Structure Matters: Tackling the Semantic Discrepancy in Diffusion Models for Image Inpainting

swift

Extension for Zed to support Swift

SwiftCU-example

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️