CUDA Army
Nvidia, CUDA, AI, SaaS, enteprise, optimization, algortithsm, data structures, compilers, inference, training, pipeleines, sharding, parallel, GPU, NPU, XPU, TPU, AMD, Intel, Pytorch, Jax, Cu
Description
We optimize you neural network training and inference pipelines for your target hardware.
We are specialized in Nvidia GPUs and libraries: CUDA, CuBLAS, CuTLASS, CuDNN, CuTe, NCCL, NVSHMEM. However, we can also work on other platforms and DSLs on-request.
We are an experienced team in quantization, pruning, distillation, distributed systems, parallelization, sharding, compilers and custom kernels such as Flash-Attention variants.
We also train custom models using your proprietary data while maintaining privacy and security. We have done projects on the following tasks:
•Computer Vision & Image Processing
•SLAM & Robotics Systems
•Reinforcement Learning
•Large Language Models & Fine-tuning
•3D Graphics & Rendering