Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A library for efficient similarity search and clustering of dense vectors.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
You like pytorch? You like micrograd? You love tinygrad! ❤️
⚡ A Fast, Extensible Progress Bar for Python and CLI
A playbook for systematically maximizing the performance of deep learning models.
A beautiful, simple, clean, and responsive Jekyll theme for academics
COLMAP - Structure-from-Motion and Multi-View Stereo
Real-Time SLAM for Monocular, Stereo and RGB-D Cameras, with Loop Detection and Relocalization Capabilities
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Flax is a neural network library for JAX that is designed for flexibility.
Extremely simple yet powerful header-only C++ plotting library built on the popular matplotlib
Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
The devkit of the nuScenes dataset.
Optax is a gradient processing and optimization library for JAX.
Code release for ConvNeXt V2 model
PROJ - Cartographic Projections and Coordinate Transformations Library
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Lift, Splat, Shoot: Encoding Images from Arbitrary Camera Rigs by Implicitly Unprojecting to 3D (ECCV 2020)
Official GitHub repository for Argoverse dataset