ZolotukhinM / minitorchLinks
☆10Updated 4 years ago
Alternatives and similar repositories for minitorch
Users that are interested in minitorch are comparing it to the libraries listed below
Sorting:
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Hyperparameter tuning via uncertainty modeling☆47Updated last year
- Worked example of the process from Python source to CUDA kernel execution with Numba☆41Updated 9 months ago
- Python code (packaged in Docker container) to run the experiments in "A Greedy Algorithm for Quantizing Neural Networks" by Eric Lybrand …☆20Updated 4 years ago
- ☆12Updated 4 years ago
- XLA integration of Open Neural Network Exchange (ONNX)☆19Updated 6 years ago
- ☆20Updated last year
- Deep learning with a multiplication budget☆47Updated 6 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Memory Optimizations for Deep Learning (ICML 2023)☆64Updated last year
- ☆42Updated 5 years ago
- Benchmarks to capture important workloads.☆31Updated 4 months ago
- ColTraIn HBFP Training Emulator☆16Updated 2 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- A cryptographically-secure pseudo-random number generator for JAX based on the 20 round ChaCha cipher.☆12Updated last year
- Document parameters using comments☆10Updated 3 years ago
- ☆22Updated this week
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- benchmarking some transformer deployments☆26Updated 2 years ago
- MCMC methods for neural networks☆10Updated last year
- Benchmarking PyTorch 2.0 different models☆21Updated 2 years ago
- ☆16Updated 2 years ago
- Tools for JAX☆47Updated last week
- Jax SSM Library☆49Updated 2 years ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 5 years ago