lanlin's repositories
cuda-samples
Samples for CUDA Developers which demonstrates features in CUDA Toolkit
flax
Flax is a neural network library for JAX that is designed for flexibility.
framework-determinism
Providing determinism in deep learning frameworks
jraph
A Graph Neural Network Library in Jax
pull-request-tutorial
An example repository for the Pull Request tutorial
python-lsp-server
Fork of the python-language-server project, maintained by the Spyder IDE team and the community
singa
a distributed deep learning platform
tensorflow
An Open Source Machine Learning Framework for Everyone
TensorRT-LLM
TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.
wandb
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.