shashankprasanna / pytorch-examples
A repository of PyTorch example
☆9Updated last year
Alternatives and similar repositories for pytorch-examples:
Users that are interested in pytorch-examples are comparing it to the libraries listed below
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.☆13Updated 2 years ago
- Shows how to do parameter ensembling using differential evolution.☆10Updated 3 years ago
- NASRec Weight Sharing Neural Architecture Search for Recommender Systems☆29Updated last year
- A dashboard for exploring timm learning rate schedulers☆19Updated 4 months ago
- PyTorch Implementation of the paper "MM1: Methods, Analysis & Insights from Multimodal LLM Pre-training"☆23Updated 2 weeks ago
- This repository holds files and scripts for incorporating simple CI/CD practices for model training in ML.☆21Updated 3 years ago
- Course repository for the Spring 2023 COMP664 course "Deep Learning" at UNC☆14Updated last year
- Contains Colab Notebooks show cool use-cases of different GCP ML APIs.☆10Updated 4 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 5 months ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆48Updated 3 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Updated 3 years ago
- PyTorch implementation of GLOM☆21Updated 2 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- ☆18Updated 2 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆48Updated last week
- Various transformers for FSDP research☆37Updated 2 years ago
- ☆15Updated 3 years ago
- Projects completed under LinuxWorld Informatics Ltd. - MLOps Training.☆12Updated 4 years ago
- Implementation of N-Grammer in Flax☆17Updated 2 years ago
- Official pytorch code for "APP: Anytime Progressive Pruning" (DyNN @ ICML, 2022; CLL @ ACML, 2022, SNN @ ICML, 2022 and SlowDNN 2023)☆16Updated 2 years ago
- Official PyTorch implementation of RIO☆18Updated 3 years ago
- Make triton easier☆47Updated 9 months ago
- Article about deploying machine learning models using grpc, pytorch and asyncio☆28Updated 2 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 2 years ago
- ☆28Updated last year
- Pixel Parsing. A reproduction of OCR-free end-to-end document understanding models with open data☆21Updated 8 months ago
- Simple and easy stable diffusion inference with LightningModule on GPU, CPU and MPS (Possibly all devices supported by Lightning).☆17Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 10 months ago
- ML/DL Math and Method notes☆59Updated last year