Srijan Keshri

@arcusbuilds · User

GitHub profile ↗ · Compare

1 followers67 repositories

Repositories

arcusbuilds/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

★ 0Forks 0

arcusbuilds/FlagGems

FlagGems is an operator library for large language models implemented in the Triton Language.

★ 0Forks 0

arcusbuilds/dynamo

A Datacenter Scale Distributed Inference Serving Framework

★ 0RustForks 0

arcusbuilds/xgboost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

★ 0Forks 0

arcusbuilds/TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

★ 0Forks 0

arcusbuilds/SpargeAttn

[ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.

★ 0Forks 0

arcusbuilds/DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

★ 0Forks 0

arcusbuilds/Trixi.jl

Trixi.jl: Adaptive high-order numerical simulations of conservation laws in Julia

★ 0Forks 0

arcusbuilds/trl

Train transformer language models with reinforcement learning.

★ 0Forks 0

arcusbuilds/cutile-rs

cuTile Rust provides a safe, tile-based kernel programming DSL for the Rust programming language. It features a safe host-side API for passing tensors to asynchronously executed kernel functions.

★ 0Forks 0

arcusbuilds/onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

★ 0Forks 0

arcusbuilds/burn

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

★ 0Forks 0