CUDA Army

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