Autoresearch for GPU kernels. Give it any PyTorch model, go to sleep, wake up to optimized Triton kernels.

Toolkit for grammars

Embeddable Postgres with real-time, reactive bindings.

Extending JAX with custom C++ and CUDA code

Library for reading and processing ML training data.

Update arXiv papers about Spiking Neural Networks daily.

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling serv…

PyGRANSO: A PyTorch-enabled port of GRANSO with auto-differentiation

A High Level API for Deep Learning in JAX

A collection of graph data used for semi-supervised node classification.

TensorFlow 2.0 library for distributed training, evaluation, model selection, and fast prototyping.

Pytorch framework for doing deep learning on point clouds.

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

A Hyperparameter Tuning Library for Keras

Static Type Checker for Python

PartNet Dataset Official Release Repo

Fast kd-tree implementation in Python

Data sets for machine learning in Python

ABC: A Big CAD Model Dataset For Geometric Deep Learning

Python library for loading and using triangular meshes.

Semantic3D segmentation with Open3D and PointNet++

TensorFlow Graphics: Differentiable Graphics Layers for TensorFlow

PointCNN: Convolution On X-Transformed Points (NeurIPS 2018)

Permutohedral Lattice C++/CUDA implementation + TensorFlow Op (CPU/GPU)

DIRT: a fast differentiable renderer for TensorFlow

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Distributed Asynchronous Hyperparameter Optimization in Python

Dynamic parser combinators in Dart.

Dart bindings for Tensorflow.