louisdewar / doxa
The OLD version of DOXA (for the new version see https://doxaai.com) A customisable platform for creating fully automated AI 🧠 competitions.
☆14Updated 2 years ago
Alternatives and similar repositories for doxa:
Users that are interested in doxa are comparing it to the libraries listed below
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆329Updated this week
- Use Jax functions in Pytorch☆231Updated last year
- Sequence Modeling with Structured State Spaces☆61Updated 2 years ago
- Running Jax in PyTorch Lightning☆83Updated last month
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆73Updated 5 months ago
- 🧱 Modula software package☆132Updated this week
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆302Updated last month
- ☆153Updated last year
- Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).☆187Updated last week
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆148Updated last month
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- ☆146Updated last month
- supporting pytorch FSDP for optimizers☆75Updated last month
- ☆545Updated last month
- ☆111Updated last week
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 6 months ago
- Open source code for EigenGame.☆29Updated last year
- LoRA for arbitrary JAX models and functions☆135Updated 10 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆255Updated last week
- Pytorch-like dataloaders in JAX.☆67Updated 3 months ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆240Updated last year
- JMP is a Mixed Precision library for JAX.☆189Updated last month
- WandB sweeps integration with Hydra sweeper☆47Updated 11 months ago
- Easy Hypernetworks in Pytorch and Jax☆96Updated last year
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago
- Efficient optimizers☆145Updated this week
- JAX Synergistic Memory Inspector☆164Updated 6 months ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago