PhD Student in Computer Vision and Autonomous Driving
- Stuttgart, Germany
- https://youtube.com/c/ftrobotics
Highlights
- Pro
Stars
Machine Learning
8 repositories
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Flax is a neural network library for JAX that is designed for flexibility.
Lightning-fast serving engine for any AI model of any size. Flexible. Easy. Enterprise-scale.
Fast and memory-efficient exact attention
Simple, safe way to store and distribute tensors