Skip to content

@augmentcode augmentcode

Change the repository type filter

All

    Repositories list

    11 repositories

    • TransformerEngine

      Public
      A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
      Python
      •
      Apache License 2.0
      •333•0•0•0•Updated Oct 21, 2024Oct 21, 2024

    • environments

      Public
      Determined AI public environments
      C
      •
      Apache License 2.0
      •41•0•0•0•Updated Oct 18, 2024Oct 18, 2024

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.4k•0•0•0•Updated Oct 16, 2024Oct 16, 2024

    • docker-credential-gcr-v2

      Public
      A Docker credential helper for GCR users
      Go
      •
      Apache License 2.0
      •93•0•0•0•Updated May 14, 2024May 14, 2024

    • apex

      Public
      A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.4k•0•0•0•Updated May 11, 2024May 11, 2024

    • rules_jsonnet

      Public
      Jsonnet rules for Bazel
      Starlark
      •
      Apache License 2.0
      •73•0•0•0•Updated Mar 28, 2024Mar 28, 2024

    • go-jsonnet

      Public
      Go
      •
      Apache License 2.0
      •236•0•0•0•Updated Mar 28, 2024Mar 28, 2024

    • xgboost

      Public
      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
      C++
      •
      Apache License 2.0
      •8.7k•0•0•0•Updated Mar 12, 2024Mar 12, 2024

    • DeeperSpeed

      Public
      DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
      Python
      •
      MIT License
      •4.1k•1•0•0•Updated Mar 5, 2024Mar 5, 2024

    • automatic-pull-request-review

      Public
      TypeScript
      •
      MIT License
      •18•0•0•0•Updated May 31, 2023May 31, 2023

    • spark

      Public
      Apache Spark - A unified analytics engine for large-scale data processing
      Scala
      •
      Apache License 2.0
      •28k•0•0•0•Updated May 5, 2023May 5, 2023