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