Skip to content
Change the repository type filter

All

    Repositories list

    • Accepted at ICML 2024 Spotlight. This code is the official implementation for the paper, "Memorization Through the Lens of Curvature of Loss Function Around Samples"
      Jupyter Notebook
      MIT License
      1610Updated Aug 1, 2024Aug 1, 2024
    • TREND

      Public
      This repository is the official implementation of the paper "TREND: Transferability based Robust ENsemble Design".
      Python
      GNU General Public License v3.0
      0100Updated Sep 21, 2023Sep 21, 2023
    • Python
      1000Updated May 1, 2023May 1, 2023
    • Codes for Structured Learning for Action Recognition in Videos
      Python
      1000Updated Oct 12, 2022Oct 12, 2022
    • Complexity-aware adaptive training and inference for distributed AI systems
      Python
      Apache License 2.0
      1000Updated Oct 7, 2022Oct 7, 2022
    • Python
      1000Updated Sep 30, 2022Sep 30, 2022
    • Python
      MIT License
      0000Updated Nov 4, 2021Nov 4, 2021
    • TSC-SNN

      Public
      0000Updated Sep 29, 2021Sep 29, 2021
    • qgm-sgp

      Public
      Python
      0000Updated Aug 17, 2021Aug 17, 2021
    • Lua
      0000Updated Jun 28, 2021Jun 28, 2021
    • Gradual Channel Pruning using Feature Relevance Scores for Convolutional Neural Networks
      Python
      3500Updated Jan 5, 2021Jan 5, 2021
    • ADV-TRAIN

      Public
      This is a framework built on top of pytorch to make machine learning training and inference tasks easier. Along with that it also enables easy dataset and network instantiations, visualize boundaries and more.
      Python
      GNU General Public License v3.0
      3100Updated Oct 23, 2020Oct 23, 2020
    • Python
      MIT License
      17000Updated Sep 17, 2020Sep 17, 2020
    • PUMA Compiler
      C++
      Other
      23101Updated Aug 4, 2020Aug 4, 2020
    • PUMA-simulator provides a detailed simulation model of a dataflow architecture built with NVM (non-volatile memory), and runs ML models compiled using the puma compiler.
      Python
      MIT License
      46300Updated Jul 14, 2020Jul 14, 2020
    • Python
      GNU Affero General Public License v3.0
      2000Updated Jun 26, 2020Jun 26, 2020
    • Training spiking networks with hybrid ann-snn conversion and spike-based backpropagation
      Python
      24000Updated Apr 27, 2020Apr 27, 2020
    • Python
      MIT License
      7000Updated Mar 21, 2020Mar 21, 2020
    • TreeCNN

      Public
      MATLAB
      0000Updated Mar 1, 2020Mar 1, 2020
    • MATLAB
      0400Updated Nov 12, 2019Nov 12, 2019
    • Structured clustering for memristive crossbar based neuromorphic architectures
      MATLAB
      MIT License
      6100Updated Oct 21, 2019Oct 21, 2019
    • GPU and CPU measurements for ML inference workloads for power, latency and throughput
      Lua
      3100Updated Aug 1, 2018Aug 1, 2018