knagrecha Goto Github PK

adaptdl

Resource-adaptive cluster scheduler for deep learning training.

analytics-zoo

Distributed Tensorflow, Keras and PyTorch on Apache Spark/Flink & Ray

best-readme-template

An awesome README template to jumpstart your projects!

cerebro-system

Data System for Optimized Deep Learning Model Selection

deepspeed-pipeline

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

flexflow

A distributed deep learning framework that supports flexible parallelization strategies.

gym-building

OpenAIGym Simple Grid Env

hierarchical_kv_cache

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

hydra

Execution framework for multi-task model parallelism. Enables the training of arbitrarily large models with a single GPU, with linear speedups for multi-gpu multi-task execution.

kabirnagrecha.com

My personal blog/website.

knagrecha.github.io

Github Pages template for academic personal websites, forked from mmistakes/minimal-mistakes

largemodel_survey

Research Exam for Kabir Nagrecha at UCSD.

lookaheaddecoding-sampling

quick test of gumbel softmax sampling

mask_rcnn

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

mesh

Mesh TensorFlow: Model Parallelism Made Easier

plumbertensorflow

Tensorflow Fork to go along with the paper "Plumber: Diagnosing and Removing Performance Bottlenecks in Machine Learning Data Pipelines". See https://github.com/mkuchnik/PlumberApp for directions.

progressbar

A very simple progress bar for c++ loops

pwwbwebsite

pytorch_modelsize

Estimates the size of a PyTorch model in memory

pytorch_mp_fix

Tensors and Dynamic neural networks in Python with strong GPU acceleration

rag_playground

saturn

Saturn accelerates the training of large-scale deep learning models with a novel joint optimization approach.

st-moe-pytorch-horovod

Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch

stb

stb single-file public domain libraries for C/C++

streaming

A Data Streaming Library for Efficient Neural Network Training

tensorflow

TF_Data_Service_Nightly

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch