facebookresearch / loop_nestLinks
Loop Nest - Linear algebra compiler and code generator.
☆21Updated 3 years ago
Alternatives and similar repositories for loop_nest
Users that are interested in loop_nest are comparing it to the libraries listed below
Sorting:
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- FlexAttention w/ FlashAttention3 Support☆27Updated last year
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆46Updated 4 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆122Updated last month
- ☆52Updated last year
- Code and data for paper "(How) do Language Models Track State?"☆19Updated 6 months ago
- Customized matrix multiplication kernels☆57Updated 3 years ago
- Better bindings for Python☆19Updated 2 years ago
- Exploration into the Firefly algorithm in Pytorch☆41Updated 8 months ago
- ☆19Updated 3 years ago
- SParse AcceleRation on Tensor Architecture☆17Updated 6 months ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- code for paper "Accessing higher dimensions for unsupervised word translation"☆21Updated 2 years ago
- Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…☆20Updated 3 months ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated last year
- NumPy+Jax with named axes and an uncompromising attitude☆22Updated 7 months ago
- Experiment of using Tangent to autodiff triton☆80Updated last year
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆64Updated 6 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆74Updated 3 months ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- 🔭 interactively explore `onnx` networks in your CLI.☆26Updated last year
- ☆16Updated last year
- Pytrees + dataclasses ❤️☆67Updated 6 months ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆16Updated 4 years ago