Alternatives and similar repositories for nni:

Users that are interested in nni are comparing it to the libraries listed below

• keras-team / autokeras
  AutoML library for deep learning
  ☆9,178Updated last month
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,502Updated last month
• horovod / horovod
  Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
  ☆14,337Updated last month
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
  ☆28,807Updated last week
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,898Updated this week
• allenai / allennlp
  An open-source NLP research library, built on PyTorch.
  ☆11,782Updated 2 years ago
• microsoft / LightGBM
  A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used …
  ☆16,874Updated this week
• pytorch / vision
  Datasets, Transforms and Models specific to Computer Vision
  ☆16,468Updated this week
• pytorch / ignite
  High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
  ☆4,569Updated last week
• facebookresearch / Detectron
  FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
  ☆26,294Updated last year
• lutzroeder / netron
  Visualizer for neural network, deep learning and machine learning models
  ☆29,114Updated this week
• google / automl
  Google Brain AutoML
  ☆6,296Updated 9 months ago
• apache / mxnet
  Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…
  ☆20,790Updated last year
• activeloopai / deeplake
  Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stre…
  ☆8,298Updated this week
• NVIDIA / DeepLearningExamples
  State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…
  ☆13,831Updated 5 months ago
• microsoft / MMdnn
  MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Co…
  ☆5,809Updated 7 months ago
• dmlc / dgl
  Python package built to ease deep learning on graph, on top of existing DL frameworks.
  ☆13,662Updated this week
• apache / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆11,930Updated last month
• keras-team / keras
  Deep Learning for humans
  ☆62,359Updated this week
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,488Updated 5 months ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆13,881Updated 10 months ago
• skorch-dev / skorch
  A scikit-learn compatible neural network library that wraps PyTorch
  ☆5,939Updated last week
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆22,626Updated 2 months ago
• fossasia / visdom
  A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.
  ☆10,069Updated 7 months ago
• automl / auto-sklearn
  Automated Machine Learning with scikit-learn
  ☆7,686Updated last month
• BVLC / caffe
  Caffe: a fast open framework for deep learning.
  ☆34,184Updated 5 months ago
• tensorpack / tensorpack
  A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility
  ☆6,301Updated last year
• tensorflow / models
  Models and examples built with TensorFlow
  ☆77,303Updated this week
• optuna / optuna
  A hyperparameter optimization framework
  ☆11,236Updated this week
• KaimingHe / deep-residual-networks
  Deep Residual Learning for Image Recognition
  ☆6,475Updated 7 years ago