tchaton / lightning-geometricLinks
Integrate pytorch
☆46Updated 4 years ago
Alternatives and similar repositories for lightning-geometric
Users that are interested in lightning-geometric are comparing it to the libraries listed below
Sorting:
- Graph Data Augmentation Library for PyTorch Geometric☆133Updated 3 years ago
- Convert scikit-learn models to PyTorch modules☆168Updated last year
- Library that contains implementations of machine learning components in the hyperbolic space☆144Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Functional deep learning☆108Updated 3 years ago
- SEML: Slurm Experiment Management Library☆186Updated 3 weeks ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- ☆100Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Official implementation (learning part) for paper: Numerically Accurate Hyperbolic Embeddings Using Tiling-Based Models☆18Updated last year
- A library for optimization on Riemannian manifolds☆112Updated 3 weeks ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆259Updated 2 years ago
- ☆64Updated 2 years ago
- ☆214Updated 3 years ago
- Official code for the Stochastic Polyak step-size optimizer☆139Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- A library for programmatically generating equivariant layers through constraint solving☆280Updated 2 years ago
- A Machine Learning workflow for Slurm.☆152Updated 5 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆97Updated 4 years ago
- ☆70Updated 2 years ago
- Neural Turing Machines in pytorch☆48Updated 4 years ago
- A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'☆31Updated last year
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Updated 2 years ago
- ☆47Updated 3 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆311Updated last week
- Codebase for Learning Invariances in Neural Networks☆96Updated 3 years ago
- ☆153Updated 5 years ago
- ☆273Updated 2 years ago