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
46 watching
484 forks
Language: Jupyter Notebook
last commit: 1 day 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. 888
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. 288
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,572
slabban/ros2_pytorch_cuda Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment. 6
nvidia/flownet2-pytorch A deep learning framework implementing an optical flow estimation model using PyTorch. 3,154
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,020
pytorch/torchtitan A native PyTorch library for large-scale language model training with distributed training capabilities 2,690
deeplearning4j/deeplearning4j A suite of tools for building and training deep learning models using the JVM. 13,703
nvidia/megatron-lm A framework for training large language models using scalable and optimized GPU techniques 10,685
hassony2/kinetics_i3d_pytorch Transfers pre-trained I3D network weights from TensorFlow to PyTorch 529
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. 430
bulletphysics/bullet3 A C++ SDK for real-time collision detection and multi-physics simulation in various fields such as VR, games, and robotics. 12,716