Low-code framework for building custom LLMs, neural networks, and other AI models

Reimplementation of a BERT based model (Shi et al, 2019), currently the state-of-the-art for English SRL. This model implements also predicate disambiguation.

RFC document, tooling and other content related to the array API standard

A cloud-native Pipeline resource.

Fast SHAP value computation for interpreting tree-based models

spaCy pipeline component for generating spaCy KnowledgeBase Alias Candidates for Entity Linking

NVIDIA's Deep Imagination Team's PyTorch Library

Multimodal Unsupervised Image-to-Image Translation

Web-scale retrieval for knowledge-intensive NLP

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008

Lime: Explaining the predictions of any machine learning classifier

Hierarchy-Aware Global Model for Hierarchical Text Classification

Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX

An easy to use PyTorch to TensorRT converter

Mobius is an AI infrastructure platform for distributed online learning, including online sample processing, training and serving.

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Fast model deployment on any cloud 🚀

TensorFlow (Python API) implementation of Neural Style

This project leverages spotify's api and provided user playlists to create and tune a neural network model that generates song recommendations based off of song data in provided playlists.

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning.

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

Common used path planning algorithms with animations.

Differentiable architecture search for convolutional and recurrent networks

A Smart, Automatic, Fast and Lightweight Web Scraper for Python

An open source python library for automated feature engineering

Datasets, papers and books on AI & Finance.

Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase

Code associated with the paper **Two are Better Than One: Joint Entity and Relation Extraction with Table-Sequence Encoders**, at EMNLP 2020

A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Train…