☆16Jan 23, 2025Updated last year
Alternatives and similar repositories for enf-jax
Users that are interested in enf-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆26Nov 18, 2025Updated 11 months ago
- ☆12Jan 9, 2024Updated 2 years ago
- A command-line tool to capture hierarchical relationships at scale☆20Updated this week
- Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/…☆18Oct 18, 2024Updated 2 years ago
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated 2 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Benchmark for learning stiff problems using physics-informed machine learning☆13Dec 15, 2021Updated 4 years ago
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago
- Graph Networks with Jax☆11Mar 23, 2022Updated 4 years ago
- ☆31May 31, 2026Updated 5 months ago
- Polynomials for ML: fast evaluation, batching, differentiation☆16Jun 17, 2026Updated 4 months ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- Tools for geometric learning☆12Sep 26, 2025Updated last year
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆16Nov 18, 2025Updated 11 months ago
- A numerical integration routine that works for the natural logarithm of functions☆12Oct 15, 2026Updated 3 weeks ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 years ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- Single neuron simulator/ prediction benchmark; uses Hodgkin-Huxley with novel mechanisms☆49Aug 12, 2026Updated 2 months ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆16Jun 24, 2023Updated 3 years ago
- ☆14Aug 8, 2023Updated 3 years ago
- ☆30Feb 16, 2023Updated 3 years ago
- ☆41Nov 14, 2022Updated 3 years ago
- Typed, annotated vectors for well-documented datasets☆12Apr 15, 2026Updated 6 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆15Nov 14, 2022Updated 3 years ago
- Bridge your scikit-learn scaler parameters between Python sessions and users. Distribute your scaling across multiple processes and data …☆12Updated this week
- Lorentz group equivariant autoencoders based on Lorentz Group Network☆11Jul 30, 2026Updated 3 months ago
- ☆16Jul 8, 2024Updated 2 years ago
- NoLiES (non-linear embeddings surveyor) is an interactive web app that helps you explore and understand non-linear embeddings of tabular …☆10Jul 1, 2021Updated 5 years ago
- Example code snipped to visualize a neural network fitting a surface to random points in space☆12Dec 26, 2021Updated 4 years ago
- The code for the paper "Editable Physically-based Reflections in Raytraced Gaussian Radiance Fields"☆69Dec 3, 2025Updated 11 months ago
- Convenience wrapper for PyAMG library☆10Feb 23, 2026Updated 8 months ago
- Fork of Flame repo for training of some new stuff in development☆20Aug 27, 2026Updated 2 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆12Nov 5, 2025Updated last year
- A domain specific language for creating scientific pipelines☆14Updated this week
- E(3) Steerable Graph Neural Network☆128Feb 10, 2023Updated 3 years ago
- Meeting repo for likelihood free inference meeting☆14Oct 6, 2022Updated 4 years ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- Package of useful sampling algorithms written in MLX.☆18Feb 27, 2024Updated 2 years ago
- ☆14Dec 15, 2022Updated 3 years ago