"PyTorch in Rust"
☆17Feb 13, 2024Updated 2 years ago
Alternatives and similar repositories for rustorch
Users that are interested in rustorch are comparing it to the libraries listed below
Sorting:
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated last year
- Implementation of Hyena Hierarchy in JAX☆10Apr 30, 2023Updated 2 years ago
- ☆32Jan 7, 2024Updated 2 years ago
- ☆14Sep 25, 2025Updated 5 months ago
- Official code repo for paper "Great Memory, Shallow Reasoning: Limits of kNN-LMs"☆23Apr 30, 2025Updated 10 months ago
- NumPy+Jax with named axes and an uncompromising attitude☆23Mar 4, 2025Updated last year
- Visualising Losses in Deep Neural Networks☆16Jul 17, 2024Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆93Jan 25, 2024Updated 2 years ago
- a simple variational auto encoder with some exploration☆12Nov 22, 2024Updated last year
- ☆23Jan 27, 2025Updated last year
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- Persistent dense gemm for Hopper in `CuTeDSL`☆15Aug 9, 2025Updated 7 months ago
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- ☆12Sep 1, 2023Updated 2 years ago
- ☆12Mar 13, 2023Updated 3 years ago
- 用C++和Python实现从头实现一个深度学习训练框架☆12Nov 22, 2020Updated 5 years ago
- ☆11Apr 5, 2021Updated 4 years ago
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆21Updated this week
- ☕️ A vscode extension for netron, support *.pdmodel, *.nb, *.onnx, *.pb, *.h5, *.tflite, *.pth, *.pt, *.mnn, *.param, etc.☆14Jun 4, 2023Updated 2 years ago
- This repo holds the research projects of our lab.☆11Jan 20, 2024Updated 2 years ago
- Stable Diffusion in TensorRT 8.5+☆14Mar 19, 2023Updated 3 years ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆17Jun 3, 2024Updated last year
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- ☆13Mar 27, 2023Updated 2 years ago
- a single-header math library☆17Nov 7, 2025Updated 4 months ago
- ☆21May 3, 2025Updated 10 months ago
- Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.☆23Jun 5, 2025Updated 9 months ago
- [EMNLP 2023]Context Compression for Auto-regressive Transformers with Sentinel Tokens☆25Nov 6, 2023Updated 2 years ago
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 7 months ago
- OneFlow Diffusers Web UI☆11Apr 11, 2023Updated 2 years ago
- Neural Networks Fundamentals with Python – implementing neural networks from scratch☆18Apr 13, 2023Updated 2 years ago
- RISC-V emulator in Zig☆15Nov 4, 2023Updated 2 years ago
- GPTQ inference Triton kernel☆321May 18, 2023Updated 2 years ago
- An UI for biographical timelines☆22Jun 3, 2025Updated 9 months ago
- [ACM Multimedia 2021] Spatiotemporal Inconsistency Learning for DeepFake Video Detection☆11Jul 13, 2023Updated 2 years ago
- TiledLower is a Dataflow Analysis and Codegen Framework written in Rust.☆13Nov 23, 2024Updated last year
- Syllable Segmentation and Cross-Lingual Generalization in a Visually Grounded, Self-Supervised Speech Model☆34Aug 27, 2023Updated 2 years ago
- SGLang is a fast serving framework for large language models and vision language models.☆20Updated this week
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆19Updated this week