Vaibhav Balloli's repositories
nfnets-pytorch
NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
Project592
EECS 592 Course Project code
vballoli.github.io
Personal Academic Website
DT4SG
EECS 602 Course Project
flower
Flower - A Friendly Federated Learning Research Framework
backpack
BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
bayesian-torch
A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
capsa-core
A data- and model-agnostic neural network wrapper for risk-aware decision making.
Chanakya
Learning Runtime Decisions for Adaptive Real-Time Perception
dm-haiku
JAX-based neural network library
flax
Flax is a neural network library for JAX that is designed for flexibility.
ignite
High-level library to help with training neural networks in PyTorch
or-gym
Environments for OR and RL Research
os-assignment
OS Assignment
pytorch-cifar
95.47% on CIFAR10 with PyTorch
pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
research-papers
Tracking and summarizing research papers I read, mostly related to machine learning
research-recap
Repository that hosts my main research notes
sample-factory
High throughput synchronous and asynchronous reinforcement learning
simulate
🎢 Creating and sharing simulation environments for embodied and synthetic data research
swift-apis
Swift for TensorFlow Deep Learning Library
swift-models
Models and examples built with Swift for TensorFlow
vision
Datasets, Transforms and Models specific to Computer Vision
VLog
A quick demo to show how to redirect GitHub Pages