Pinned Repositories
awesome-athletes
⚾ machine learning meets sports analytics: clustering similar athletes, gauging performance & injury rates!
ECE-239AS-Project
RNNs on EEG Data
ml-algorithms
💻 my own machine learning library with algorithms implemented from scratch
neuralnets
🤖 bare-bones implementation of a neural network with numpy
nn-init-demo
Demos demonstrations of things that can go wrong when initializing weights
paper-analysis
Analysis & Implementation of deep learning papers
scripts
Some productivity scripts
word2vec
🎬 machine learning on movie reviews with word embeddings
distbelief
Implementing Google's DistBelief paper
rohan-varma's Repositories
rohan-varma/scripts
Some productivity scripts
rohan-varma/pt-shard-experiments
rohan-varma/rohan-varma.github.io
Software Engineer @ Facebook
rohan-varma/torchtune-2
A Native-PyTorch Library for LLM Fine-tuning
rohan-varma/ao
The torchao repository contains api's and workflows for quantization and pruning gpu models.
rohan-varma/d2go
D2Go is a toolkit for efficient deep learning
rohan-varma/DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
rohan-varma/drlm_pt_ddp
rohan-varma/fairscale
PyTorch extensions for high performance and large scale training.
rohan-varma/functional_ddp
rohan-varma/gloo
Collective communications library with various primitives for multi-machine training.
rohan-varma/llama-recipes
Examples and recipes for Llama model
rohan-varma/llm_inference
rohan-varma/metaseq
Repo for external large-scale work
rohan-varma/mixprecision
modular repo to work on bert attn issue with mixed precision
rohan-varma/multimodal
TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.
rohan-varma/pt_build_scripts
rohan-varma/ptd_benchmark
rohan-varma/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
rohan-varma/t5_11
housing our model example of fine tuning an 11B t5 with FSDP
rohan-varma/tnt
A lightweight library for PyTorch training tools and utilities
rohan-varma/torch-script
rohan-varma/torchdistx
Torch Distributed Experimental
rohan-varma/torchrec
Pytorch domain library for recommendation systems
rohan-varma/torchrec-1
Pytorch domain library for recommendation systems
rohan-varma/torchtune
A Native-PyTorch Library for LLM Fine-tuning
rohan-varma/transformer_framework
framework for plug and play of various transformers (vision and nlp) with FSDP
rohan-varma/transformer_nuggets
A place to store reusable transformer components of my own creation or found on the interwebs
rohan-varma/tutorials
PyTorch tutorials.
rohan-varma/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)