jetson-containers

AI/ML container framework

Provides modular containers for AI/ML development on NVIDIA Jetson devices

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

GitHub

2k stars
47 watching
489 forks
Language: Jupyter Notebook
last commit: about 1 month ago
Linked from 1 awesome list

containersdockerdockerfilesjetsonmachine-learningnumpynvidiapandaspytorchros-containersros2-foxyscikit-learntensorflow

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
dusty-nv/jetson-reinforcement Provides tools and libraries for training intelligent agents to perform complex tasks in robotics using deep reinforcement learning and GPU acceleration. 891
rbonghi/jetson_easy Automates the setup and configuration of NVIDIA Jetson boards by updating and patching the kernel, installing ROS, and setting up other necessary components. 290
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,978
slabban/ros2_pytorch_cuda Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment. 7
nvidia/flownet2-pytorch A deep learning framework implementing an optical flow estimation model using PyTorch. 3,159
trn84/recipe-wizard Automates creation of Docker containers for running OpenGL and CUDA applications on remote servers with optional integration of ROS, Gazebo, and Python deep learning tools. 40
nvlabs/imaginaire A PyTorch library providing optimized implementations of image and video synthesis methods developed at NVIDIA. 4,025
pytorch/torchtitan A native PyTorch library for training large language models using distributed parallelism and optimization techniques. 2,765
deeplearning4j/deeplearning4j A suite of tools for building and training deep learning models using the JVM. 13,718
nvidia/megatron-lm A framework for training large language models using scalable and optimized GPU techniques 10,804
hassony2/kinetics_i3d_pytorch Transfers pre-trained I3D network weights from TensorFlow to PyTorch 532
ramon-oliveira/aorun A deep learning framework on top of PyTorch for building neural networks. 61
randl/shufflenetv2-pytorch An implementation of a lightweight convolutional neural network architecture for mobile devices 191
ericsun99/shufflenet-v2-pytorch A PyTorch implementation of a ShuffleNet-v2 neural network architecture for image classification. 431
bulletphysics/bullet3 A C++ SDK for real-time collision detection and multi-physics simulation in various fields such as VR, games, and robotics. 12,777