conda-forge / jaxlib-feedstock
A conda-smithy repository for jaxlib.
☆16Updated 2 months ago
Alternatives and similar repositories for jaxlib-feedstock:
Users that are interested in jaxlib-feedstock are comparing it to the libraries listed below
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- LaTeX source code for the slides☆23Updated 3 years ago
- Multiplicative Normalizing Flows in PyTorch.☆24Updated last month
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 8 months ago
- Large-Scale Machine and Deep Learning in PyTorch.☆21Updated 3 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated last year
- ☆11Updated 9 months ago
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- Quantification of Uncertainty with Adversarial Models☆28Updated last year
- Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)☆22Updated 2 years ago
- Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen☆25Updated 4 years ago
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 3 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- ☆37Updated 2 years ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 10 months ago
- High-performance tokenized language data-loader for Python C++ extension☆12Updated 6 months ago
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆72Updated last year
- Repository for the code assignment of the Deep Learning 1 course, Fall 2021 edition☆10Updated 2 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- Isomorphisms of quiver representations applied to neural networks.☆16Updated 6 months ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆45Updated 2 weeks ago
- Convolutions and more as einsum for PyTorch☆14Updated 8 months ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 2 years ago
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆44Updated 2 years ago