A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

Fine-tune mistral-7B on 3090s, a100s, h100s

text and image to video generation: CogVideoX (2024) and CogVideo (ICLR 2023)

The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

A PyTorch-based framework for Quantum Classical Simulation, Quantum Machine Learning, Quantum Neural Networks, Parameterized Quantum Circuits with support for easy deployments on real quantum compu…

Position Tracking With IMU

Official code repository of "HyperDiffusion: Generating Implicit Neural Fields with Weight-Space Diffusion" @ ICCV 2023

Foundational Models for State-of-the-Art Speech and Text Translation

[ICCV2023] IntrinsicNeRF: Learning Intrinsic Neural Radiance Fields for Editable Novel View Synthesis

Doctor Dignity is an LLM that can pass the US Medical Licensing Exam. It works offline, it's cross-platform, & your health data stays private.

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

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

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

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

Stable Diffusion with Core ML on Apple Silicon

CLI platform to experiment with codegen. Precursor to: https://lovable.dev

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable…

Official Code for DragGAN (SIGGRAPH 2023)

MONAI Generative Models makes it easy to train, evaluate, and deploy generative models and related applications

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

Virtual whiteboard for sketching hand-drawn like diagrams

[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A tool for exploring each layer in a docker image

Convert Machine Learning Code Between Frameworks

Implementation for the works "VecGAN: Image-to-Image Translation with Interpretable Latent Directions" (ECCV 2022) and "Face Attribute Editing with Disentangled Latent Vectors"

Explanation to key concepts in ML

🦜🔗 Build context-aware reasoning applications

AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/