amirabbasasadi / RockyMLLinks
⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems
☆19Updated 2 years ago
Alternatives and similar repositories for RockyML
Users that are interested in RockyML are comparing it to the libraries listed below
Sorting:
- A Gentle Principled Introduction to Deep Reinforcement Learning☆19Updated 7 months ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- This repository provides a Colab Notebook that shows how to use Spatial Transformer Networks inside CNNs in Keras.☆37Updated 3 years ago
- Experimental version of jxbz/agd implementing support for bias terms, affine parameters, transformers, etc.☆12Updated 2 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Updated 4 years ago
- A neural network hyper parameter tuner☆30Updated last year
- Datasets associated with pyprobml☆20Updated 2 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- An attempt at the implementation of GLOM, Geoffrey Hinton's paper for emergent part-whole hierarchies from data☆37Updated 4 years ago
- Yaae: Yet another autodiff engine (written in Numpy).☆28Updated 2 years ago
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- A tutorial on JAX (https://github.com/google/jax/)☆47Updated 6 years ago
- Jax SSM Library☆48Updated 2 years ago
- Quantification of Uncertainty with Adversarial Models☆29Updated 2 years ago
- Companion code for a tutorial on using Hydra.☆30Updated 4 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 3 years ago
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- ☆20Updated 2 years ago
- List of awesome JAX resources☆13Updated 2 years ago
- ☆18Updated last year
- Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)☆22Updated 3 years ago
- Deep Critical Learning. Implementation of ProSelfLC, IMAE, DM, etc.☆31Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆82Updated 2 years ago
- Benchmarking PyTorch 2.0 different models☆20Updated 2 years ago
- ☆37Updated 3 years ago
- Data Twinning☆25Updated 2 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆59Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago