Abhay Rawat's starred repositories
pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
PlotNeuralNet
Latex code for making neural networks diagrams
PyTorch-GAN
PyTorch implementations of Generative Adversarial Networks.
ipex-llm
Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, Baichuan, Mixtral, Gemma, Phi, etc.) on Intel CPU and GPU (e.g., local PC with iGPU, discrete GPU such as Arc, Flex and Max); seamlessly integrate with llama.cpp, Ollama, HuggingFace, LangChain, LlamaIndex, DeepSpeed, vLLM, FastChat, Axolotl, etc.
pointnet.pytorch
pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593
CelebAMask-HQ
A large-scale face dataset for face parsing, recognition, generation and editing.
robotics-toolbox-python
Robotics Toolbox for Python
lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
kaolin-wisp
NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
Dassl.pytorch
A PyTorch toolbox for domain generalization, domain adaptation and semi-supervised learning.
com.unity.perception
Perception toolkit for sim2real training and validation in Unity
photometric_optimization
Photometric optimization code for creating the FLAME texture space and other applications
pybullet-robot-envs
A Python package that collects robotic environments based on the PyBullet simulator, suitable to develop and test Reinforcement Learning algorithms on simulated grasping and manipulation applications.
EfficientNeRF
The official code for "Efficient Neural Radiance Fields" in CVPR2022.
AnalyticMesh
An Efficient Implementation of Analytic Mesh Algorithm for 3D Iso-surface Extraction from Neural Networks
analytics-zoo
Distributed Tensorflow, Keras and BigDL on Apache Spark